leetcode 563 二叉树的坡度

https://leetcode-cn.com/problems/binary-tree-tilt/

题目

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

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

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

思路1

这个算是非常经典的二叉树递归练习题了，直接对问题的定义建模就好。

首先，求左子树的节点之和，那么有：

int sumLeft = dfs(node->left);

右子树同理：

int sumRight = dfs(node->right);

差的绝对值：

ans += abs(sumLeft - sumRight);

最后有：

int ans = 0;

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

int dfs(TreeNode* node) {
    if (node == nullptr) {
        return 0;
    }
    int sumLeft = dfs(node->left);
    int sumRight = dfs(node->right);
    ans += abs(sumLeft - sumRight);
    return sumLeft + sumRight + node->val;
} 

本题就相当于是一个"求二叉树所有节点和"的简单改版。