979. Distribute Coins in Binary Tree

这题很有意思，本来认为树的题目没有什么难的了，但是这题确实想了很久也没有做出来。

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;
    }
};

题目不需要求出具体最后的状态，我们知道最后肯定会达到理想的状态，所以我们只要记住流入流出就行了。