1.题目
这道题使2024-2-28的签到题,题目难度为中等。
考察知识点:贪心 + 满二叉树
题目链接:2673. 使二叉树所有路径值相等的最小代价 - 力扣(LeetCode)
2.思路
在做这道题之前我们得知道满二叉树是什么?对于满二叉树,除了最后一层的叶子节点,其它所有结点都有两个子结点,下面是一个对比图:
有了这个概念之后我们再来看题目,题目给我们一个cost数组来表示整棵树的值,这里我们可以根据满二叉树的特性来推出一个结论:
在cost数组中,父结点的下标索引如果为x,则它的左子结点下标索引为2 * x + 1,右子节点的下标索引为2 * x + 2。(uu们可以自己带入验证一下)
对于这道题,我们肯定首先想到的算法是贪心算法,那么我们如何贪心呢?对于整棵树,如果我们从上往下遍历修改对我们的目标是十分不利的,为啥?上游影响力肯定大于下游影响力,同样,如果我们修改了树中靠上面的结点值,那么对于下面的所有路径,它们的路径是不是都会跟着改变,而这只会造成更改次数的浪费。举个例子,如果我们刚开始直接更改根节点的值100次,是不是就是浪费了100次,因此我们这道题需要自下往上的修改。
那么如果自下往上怎么修改呢?我们只需要找到两个相邻的兄弟结点,它俩的差值是不是就是我们本次的修改次数呢?然后我们再将它们两个较大的那个值加到它的父节点值里面,这样就可以代替这两条子路径的值了,然后一直重复前面的步骤,先比较相邻兄弟结点,然后将值加到父节点里面。
3.代码
class Solution:
def minIncrements(self, n: int, cost: List[int]) -> int:
# 累加次数
rst = 0
# 从后往前遍历,这里相当于自下往上
# 这里步长为-2,是因为我们只需要比较兄弟结点,因此这里的i都是左结点下标索引
for i in range(n-2,0,-2):
# 两个兄弟结点的差值即为本次修改的次数
rst += abs(cost[i] - cost[i+1])
# 将两个兄弟结点的大值累加到父母结点里面
cost[i//2] += max(cost[i],cost[i+1])
return rst