数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济
心理学
社会研究
时尚研究
法律研究C++ 中分发二叉树中的硬币
假设我们有一个具有 N 个节点的二叉树的根，其中树中的每个节点都具有 node.val 个硬币，并且共有 N 个硬币。在一场移动中，我们可以选择两个相邻的节点，并且只将一个硬币从一个节点移动到另一个节点。（该移动可以从父节点到子节点，或从子节点到父节点。）我们必须找出使每个节点都恰好具有一个硬币所需的移动次数。
因此，如果树像这样−
那么输出将为 3。从左孩子处，向根发送 2 个硬币（每个硬币移动一次，所以总共移动 2 次），然后将一个硬币从根移动到右孩子，因此总共有 3 次移动。
为了解决此问题，我们将遵循以下步骤 −
定义一个名为 solve() 的递归方法，该方法将采用一个名为 root 的节点
如果 root 为空，则返回 0
l := solve(root 的左孩子)
r := solve(root 的右孩子)
ans := |l| + |r|
返回 l + r + root 的值 – 1
在 main 部分，设置 ans := 0，调用 solve(root)，然后返回 ans
示例
#include <bits/stdc++.h>
using namespace std;
class TreeNode{
public:
int val;
TreeNode *left, *right;
TreeNode(int data){
val = data;
left = NULL;
right = NULL;
}
};
void insert(TreeNode **root, int val){
queue<TreeNode*> q;
q.push(*root);
while(q.size()){
TreeNode *temp = q.front();
q.pop();
if(!temp->left){
if(val != NULL)
temp->left = new TreeNode(val);
else
temp->left = new TreeNode(0);
return;
}else{
q.push(temp->left);
}
if(!temp->right){
if(val != NULL)
temp->right = new TreeNode(val);
else
temp->right = new TreeNode(0);
return;
}else{
q.push(temp->right);
}
}
}
TreeNode *make_tree(vector<int> v){
TreeNode *root = new TreeNode(v[0]);
for(int i = 1; i<v.size(); i++){
insert(&root, v[i]);
}
return root;
}
class Solution {
public:
int ans;
int solve(TreeNode* root){
if(!root)return 0;
int l = solve(root->left);
int r = solve(root->right);
ans += abs(l) + abs(r);
return l + r + root->val - 1;
}
int distributeCoins(TreeNode* root) {
ans = 0;
solve(root);
return ans;
}
};
main(){
vector<int> v = {0,3,0};
TreeNode *root = make_tree(v);
Solution ob;
cout << (ob.distributeCoins(root));
} 输入
[0,3,0]
输出
3
更新于： 02-May-2020
183 次浏览
广告