สมมติว่าเรามีรากของต้นไม้ไบนารีที่มีโหนด N โดยที่แต่ละโหนดในทรีมีจำนวนเหรียญ node.val และมีทั้งหมด N เหรียญ ในการย้ายครั้งเดียว เราสามารถเลือกโหนดที่อยู่ติดกันสองโหนด และย้ายเพียงเหรียญเดียวจากโหนดหนึ่งไปยังโหนดอื่น (การย้ายอาจมาจากโหนดหลักไปยังโหนดย่อย หรือจากโหนดย่อยไปยังโหนดหลัก) เราต้องหาจำนวนการเคลื่อนไหวที่จำเป็นเพื่อให้ทุกโหนดมีเหรียญเดียว
ดังนั้นถ้าต้นไม้เป็นเหมือน −
จากนั้นผลลัพธ์จะเป็น 3 จากลูกด้านซ้าย ให้ส่ง 2 เหรียญไปที่รูท (หนึ่งย้ายสำหรับแต่ละเหรียญ ดังนั้นทั้งหมด 2 การเคลื่อนไหว) จากนั้นย้ายหนึ่งเหรียญจากรูทไปยังเด็กที่ถูกต้อง ดังนั้นทั้งหมดมี 3 การเคลื่อนไหว
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
-
กำหนดวิธีการเรียกซ้ำหนึ่งวิธีที่เรียกว่า Solve() ซึ่งจะใช้โหนดที่เรียกว่า root
-
ถ้ารูทเป็นโมฆะ ให้คืนค่า 0
-
l :=แก้ (ด้านซ้ายของรูท)
-
r :=แก้ (ทางขวาของรูท)
-
ตอบ :=|l| + |r|
-
คืนค่า l + r + ค่ารูท – 1
-
ในส่วนหลัก ให้ตั้งค่า ans :=0, call dissolve(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
