这题很有意思,本来认为树的题目没有什么难的了,但是这题确实想了很久也没有做出来。
class Solution {
public:
int distributeCoins(TreeNode* root) {
int moves = 0;
assert(dfs(root, moves) == 0);
return moves;
}
private:
int dfs(TreeNode* node, int& moves) {//返回的是为了保持node子树的平衡,需要流出或者流出的硬币。同时node子树内部已保持了平衡。
if (!node)
return 0;
int lm = dfs(node->left, moves);
int rm = dfs(node->right, moves);
moves += (abs(lm) + abs(rm));
return node->val + lm + rm - 1;
}
};
题目不需要求出具体最后的状态,我们知道最后肯定会达到理想的状态,所以我们只要记住流入流出就行了。