【Leetcode】563. 二叉树的坡度

题目

给定一个二叉树，计算整个树的坡度。

一个树的节点的坡度定义即为，该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。

整个树的坡度就是其所有节点的坡度之和。

示例:

输入: 
     1
   /   \
  2     3
输出: 1
解释: 
结点的坡度 2 : 0
结点的坡度 3 : 0
结点的坡度 1 : |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1

注意:

任何子树的结点的和不会超过32位整数的范围。
坡度的值不会超过32位整数的范围。

思路

这道题拿到的时候，我的思路是这样的：首先遍历二叉树，求出各个节点的值，然后将该节点的值保存到字典里，key为节点，value就是节点的值；然后，再次遍历节点，对每个节点的求其坡度，同时加到结果ans中，这个过程中会去取节点的值，那么之前保存的字典就用上了。

后来想了一下，之前的第二次遍历和第一次其实没有什么区别，能不能将这两次遍历合为一体，把需要做的事情同时做了呢？当然可以。思路如下：
遍历二叉树的时候，left保存左子树的和，right保存右子树的和，那么，该节点的坡度就是left-right的绝对值，把这个值直接加到最终的结果ans中，然后递归返回的是left+right+node.val，这样就表示，别人递归调用的时候，返回的是这个节点的左子树和右子树以及该节点三者的和，回头想想left和right，是不是这回事。当然，递归的出口是，node==None，返回0

代码

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

class Solution:
    def findTilt(self, root: TreeNode) -> int:
        ans = [0]
        def inorder(r):
            if(not r):
                return 0
            left = inorder(r.left)
            right = inorder(r.right)
            ans[0] += abs(left - right)
            return left + right + r.val
        inorder(root)
        return ans[0]