LeetCode 563. 二叉树的坡度

题目：【https://leetcode.cn/problems/binary-tree-tilt/description/】

给你一个二叉树的根节点 root ，计算并返回 整个树 的坡度 。
一个树的 节点的坡度 定义即为，该节点左子树的节点之和和右子树节点之和的 差的绝对值 。如果没有左子树的话，左子树的节点之和为 0 ；没有右子树的话也是一样。空结点的坡度是 0 。
整个树 的坡度就是其所有节点的坡度之和。

示例 ：

输入：root = [1,2,3]
输出：1
解释：
节点 2 的坡度：|0-0| = 0（没有子节点）
节点 3 的坡度：|0-0| = 0（没有子节点）
节点 1 的坡度：|2-3| = 1（左子树就是左子节点，所以和是 2 ；右子树就是右子节点，所以和是 3 ）
坡度总和：0 + 0 + 1 = 1

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */

class Solution {
public:
    // 返回当前节点及所有子节点之后
    int traverse(TreeNode* node, int& res) {
        if (node == nullptr) {
            return 0;
        }
        
        // 递归处理左右子树
        int sumLeft = traverse(node->left, res);
        int sumRight = traverse(node->right, res);
        
        // 结果加当前节点的坡度
        res += abs(sumLeft - sumRight);
        return sumLeft + sumRight + node->val;
    }
    int findTilt(TreeNode* root) {
        int res = 0;
        traverse(root, res);
        return res;
    }
};