木又连续日更第100天(100/100)
木又的第143篇leetcode解题报告
二叉树
类型第33篇解题报告
leetcode第563题:二叉树的坡度
https://leetcode-cn.com/problems/binary-tree-tilt
【题目】
给定一个二叉树,计算整个树的坡度。
一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。
整个树的坡度就是其所有节点的坡度之和。
示例:
输入:
1
/ \
2 3
输出: 1
解释:
结点的坡度 2 : 0
结点的坡度 3 : 0
结点的坡度 1 : |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1
注意:
任何子树的结点的和不会超过32位整数的范围。 坡度的值不会超过32位整数的范围。
【思路】
分别得到左右子树的元素和,再计算并累加坡度。
有些小伙伴或许还在疑问,递归到底怎么写呀?
其实很简单的。
主要两步:假设已经有个函数能够实现我们想要的子功能,我们要怎样调用;函数应该什么时候返回结果,返回什么结果。
比如这道题
1)假设我们有个函数能够实现求子树和的功能,那么怎么调用求得坡度呢?
# 求得左右子树和
left = fun(node.left)
right = fun(node.right)
# 求得坡度
tilt = abs(left - right)
2)什么时候返回子树和呢?
一是访问到叶子节点,返回0;二是函数结束,返回以本节点为根节点的树的和。
if not node:
return 0
left = ...
...
return left + right + node.val
最后把两段代码合在一起,解决几个问题:哪些应该是全局变量,是否还有特殊情况等。
【代码】
python版本
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def findTilt(self, root):
"""
:type root: TreeNode
:rtype: int
"""
self.tilt = 0
self.get_tilt(root)
return self.tilt
def get_tilt(self, node):
if not node:
return 0
left = self.get_tilt(node.left)
right = self.get_tilt(node.right)
print(node.val, left, right)
self.tilt += abs(left - right)
return left + right + node.val
C++版本
/**
* 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 findTilt(TreeNode* root) {
int tilt = 0;
get_tilt(root, tilt);
return tilt;
}
int get_tilt(TreeNode* node, int& tilt){
if(!node)
return 0;
int left = get_tilt(node->left, tilt);
int right = get_tilt(node->right, tilt);
tilt += abs(left - right);
return left + right + node->val;
}
};