leetcode —— 979. 在二叉树中分配硬币

给定一个有 N 个结点的二叉树的根结点 root，树中的每个结点上都对应有 node.val 枚硬币，并且总共有 N 枚硬币。

在一次移动中，我们可以选择两个相邻的结点，然后将一枚硬币从其中一个结点移动到另一个结点。(移动可以是从父结点到子结点，或者从子结点移动到父结点。)。

返回使每个结点上只有一枚硬币所需的移动次数。

示例 1：

输入：[3,0,0]
输出：2
解释：从树的根结点开始，我们将一枚硬币移到它的左子结点上，一枚硬币移到它的右子结点上。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/distribute-coins-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
————————————————
解题思路：使用深度优先遍历，创建一个函数funs(root)，该函数的作用是计算当前结点root的金币量（这时候当前节点的子节点的金币数都为1）。

当知道当前节点的左子节点的金币数量和当前节点的右子节点的金币数量，就可以知道当左子节点和右子节点的金币数为1时当前节点的金币个数和需要移动的次数。

用变量ans保存整体需要移动的金币次数。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def __init__(self):
        self.ans = 0
    def distributeCoins(self, root: TreeNode) -> int:
        def funs(root):
            if not root:
                return 0
            elif root.right and root.left:  # 如果存在左子树和右子树
                numl = funs(root.left) # 计算左子节点的金币数
                numr = funs(root.right) # 计算右子节点的禁金币数
                self.ans += abs(numl-1) + abs(numr-1) # 左子节点和右子节点的金币数为1需要移动的次数
                return root.val + numl + numr - 2 # 更新当前节点的金币数
            elif root.left: # 如果右子树为空
                num = funs(root.left) # 计算左子节点的金币数
                self.ans += abs(num-1)  # 左子节点金币数为1需要移动的次数
                return root.val + num - 1 # 更新当前节点的金币数
            elif root.right: # 如果左子树为空
                num = funs(root.right)  # 计算右子节点的金币数
                self.ans += abs(num-1)  # 右子节点金币数为1需要移动的次数
                return root.val + num - 1  # 更新当前节点的金币数
            else:
                return root.val  # 如果没有左右子树，返回当前结点的金币数
        funs(root) 
        return self.ans

该算法会遍历二叉树的每一个节点，因此时间复杂度为O(n)，需要使用堆栈，因此空间复杂度为O(H)，H为二叉树的高度。