【LeetCode】563. 二叉树的坡度

题目链接：https://leetcode-cn.com/problems/binary-tree-tilt/description/

题目描述

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

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

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

示例:

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

注意:

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

解决方法

用递归，方法类似于543. 二叉树的直径

/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ //用递归，方法类似于543. 二叉树的直径 class Solution { public: int findTilt(TreeNode* root) { int slope=0; nodeSum(root,slope); return slope; } private: int nodeSum(TreeNode* root,int &slope){ if (!root) return 0; int left_nodeSum=nodeSum(root->left,slope); int right_nodeSum=nodeSum(root->right,slope); slope+=abs(left_nodeSum-right_nodeSum); return root->val+left_nodeSum+right_nodeSum; } };