給定一個有 N 個結點的二叉樹的根結點 root,樹中的每個結點上都對應有 node.val 枚硬幣,并且總共有 N 枚硬幣。
在一次移動中,我們可以選擇兩個相鄰的結點,然后將一枚硬幣從其中一個結點移動到另一個結點。(移動可以是從父結點到子結點,或者從子結點移動到父結點。)。
返回使每個結點上只有一枚硬幣所需的移動次數。
輸入:[3,0,0]
輸出:2
解釋:從樹的根結點開始,我們將一枚硬幣移到它的左子結點上,一枚硬幣移到它的右子結點上。
代碼
/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode(int x) { val = x; }* }*/
class Solution {int ans=0;public int distributeCoins(TreeNode root) {Coins(root);return ans;}public int Coins(TreeNode root) {if(root==null) return 0;int l=Coins(root.left);//左子樹可能多出或者缺少的金幣數,用正負號區分int r=Coins(root.right);//右子樹可能多出或者缺少的金幣數ans+= Math.abs(l)+ Math.abs(r);//如果子樹缺少若干個金幣,父節點就要移動若干個金幣下去子樹。
//如果子樹多出若干個金幣,就要移動若干個金幣上去父節點。return root.val-1+l+r;//再將多出或者缺少的金幣數目返回給父節點}
}