1. 题目描述
给定一个二叉树,计算整个树的坡度。
一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。
整个树的坡度就是其所有节点的坡度之和。
示例:
输入:
1
/ \
2 3
输出: 1
解释:
结点的坡度 2 : 0
结点的坡度 3 : 0
结点的坡度 1 : |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1
注意:
- 任何子树的结点的和不会超过32位整数的范围。
- 坡度的值不会超过32位整数的范围。
2. 题解思路
该题主要是求左子树的结点之和和右子树结点之和的差的绝对值,那么首先就要求子树节点之和,之前一篇文章详细讲了节点之和怎么求https://blog.csdn.net/oyall520/article/details/104944199
(1)所以我们可以先定义一个辅助函数SumOfNode求节点之和;
(2)然后在主函数里通过调用辅助函数和调用自身求出每个节点的左右子树节点之和的绝对值
3. 第一种写法
这种写法就是上面思路的复现,但是运行时间太慢,仔细看了一下确实复杂了,我在 SumOfNode 函数里求了左右子树的节点和,然后在主函数里又调用了两次 SumOfNode 函数求节点左右子树和,来求绝对值之差;还要调用自身,可以想象有多慢!
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def findTilt(self, root: TreeNode) -> int:
if not root:
return 0
if not root.left and not root.right:
return 0
l = self.SumOfNode(root.left)
r = self.SumOfNode(root.right)
return self.findTilt(root.left) + self.findTilt(root.right) + abs(l - r)
def SumOfNode(self, root):
if not root:
return 0
leave = root.val
left = self.SumOfNode(root.left)
right = self.SumOfNode(root.right)
return left + right + leave
4. 第二种写法
由第一种写法引发的思考:为什么不直接在辅助函数里求出左右子树的节点和之后就算出两者之差的绝对值,然后用一个变量累加呢,这样可以省去好几个递归调用,速度自然就快了。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def findTilt(self, root: TreeNode) -> int:
self.ans = 0
# if not root.left and not root.right:
# return 0
# l = self.SumOfNode(root.left)
# r = self.SumOfNode(root.right)
self.SumOfNode(root)
return self.ans
def SumOfNode(self, root):
if not root:
return 0
leave = root.val
left = self.SumOfNode(root.left)
right = self.SumOfNode(root.right)
self.ans += abs(left - right)
return left + right + leave