Leetcode 563 二叉树的坡度

题目

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

一个树的 节点的坡度 定义即为，该节点左子树的节点之和和右子树节点之和的 差的绝对值 。如果没有左子树的话，左子树的节点之和为 0 ；没有右子树的话也是一样。空结点的坡度是 0 。

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

解题思路

  看到左子树和右子树节点之和，想到了递归。其次是对左右子树之和作差，那么需要在递归的过程中拿到左右子树，那就需要后序遍历。在后序遍历的过程中，求左右子树之和的差的绝对值就是这个题的解法。

代码

class Solution {
    public int ans = 0;

    public int dfs(TreeNode root) {
        if (root == null) return 0;
        int leftResult = dfs(root.left);
        int rightResult = dfs(root.right);

        ans += Math.abs(leftResult - rightResult);
        return root.val + leftResult + rightResult;
    }

    public int findTilt(TreeNode root) {
        dfs(root);
        return ans;
    }
}