使用 C++ 在二叉树中分配硬币
c++server side programmingprogramming更新于 2024/11/8 21:38:00
假设我们有一个二叉树的根，该树有 N 个节点，树中的每个节点都有 node.val 个硬币，总共有 N 个硬币。在一次移动中，我们可以选择两个相邻节点，并将一个硬币从一个节点移动到另一个节点。(移动可能是从父节点到子节点，或从子节点到父节点。)我们必须找到使每个节点恰好有一枚硬币所需的移动次数。
因此，如果树像 −
则输出将为 3。从左子节点，将 2 枚硬币发送到根节点(每枚硬币移动一次，因此总共移动 2 次)，然后将一枚硬币从根节点移动到右子节点，因此总共移动 3 次。
为了解决这个问题，我们将遵循以下步骤 −
定义一个名为solve()的递归方法，它将采用一个名为root的节点
如果root为空，则返回0
l :=solve(根节点的左侧)
r :=solve(根节点的右侧)
ans := |l| + |r|
返回 l + r + 根的值 – 1
在主部分中，设置 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
