leetcode-563. 二叉树的坡度

563. 二叉树的坡度 - 力扣（Leetcode）

坡度的计算需要4个数

1. 左子树所有节点的和
2. 右子树所有结点的和
3. 左子树的坡度
4. 右子树的坡度

左子树与右子树节点差值的绝对值为当前节点的坡度

左子树与右子树的坡度与当前节点的坡度作为第二个值返回

下面的代码使用了 go 的双返回值特性，你也可以使用一个全局变量，但是不知道为什么 leetcode 里面使用全局变量无法赋值

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func findTilt(root *TreeNode) int {
    _, slope := dfs(root)
    return slope
}

func dfs(root *TreeNode) (int, int) {
    if root == nil {
        return 0, 0
    }

    left, lSlope := dfs(root.Left)
    right, rSlope := dfs(root.Right)

    curSlope := left - right
    if curSlope < 0 {
        curSlope = -curSlope
    }

    return left + right + root.Val, curSlope + lSlope + rSlope
}