資料結構
網路
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 次瀏覽
廣告