给定一个二叉树,计算整个树的坡度。
一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。
整个树的坡度就是其所有节点的坡度之和。
示例:
输入:
1
/ \
2 3
输出:1
解释:
结点 2 的坡度: 0
结点 3 的坡度: 0
结点 1 的坡度: |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1
提示:
任何子树的结点的和不会超过 32 位整数的范围。
坡度的值不会超过 32 位整数的范围。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-tilt
最近写了好多二叉树的题目,但是碰到这样的还是不会写,分析了一下觉得应该是后序遍历,但是后序的迭代不会写,递归也分析不来,最后还是得去看答案。
一开始看官方给的答案不是很懂,然后看了评论区的一个答案,那个答案的方式有点类似二叉树的最大深度的解法,
就是维护一个递归函数,可以求每个结点所有子树加本身结点的和,然后在主函数里再用递归对每个结点遍历,并利用前面的递归函数求其左右子树的和,并求坡度。
不过这个方法对结点的遍历次数太多了,效率很低。
然后就是官方写的解法,递归版本的后序遍历,执行用时果然少了很多。不过还是不懂后序遍历怎么用迭代写,我看后面提交记录里有人用迭代解这一道题,但是又用了stack又用了set,
感觉挺复杂的,目前我功力还不够。
总是在找工作的时候才发现自己啥都不会,实在是太难了,现在才开始学习语言和数据结构也不知道来不来的及。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int dfs(TreeNode* root,int & tilt){
if(root==NULL) return 0;
int left=dfs(root->left,tilt);
int right=dfs(root->right,tilt);
tilt+=abs(left-right);
return left+right+root->val;
}
int findTilt(TreeNode* root) {
int tilt=0;
dfs(root,tilt);
return tilt;
}
};