2024.2.28
题目来源
我的题解
方法一 自底向上+贪心
- 对于任一叶结点,它的值为 x,它的兄弟节点的值为 y。可以发现,对于树上的其余节点,它们要么同时是这两个叶节点的祖先,要么同时不是这两个叶节点的祖先。对这些节点进行一次操作,要么同时增加了根到这两个叶节点的路径值 1,要么没有任何效果。因此,要想使得根到这两个叶节点的路径值相等,只能增加 x和 y本身。要想操作次数最小,就需要求|x-y|的差值,将较小的值增加到较大的值的次数。
- 对于非叶子节点,可以通过虚拟去除叶节点的方式(即将叶子节点中较大的值加入到父节点中),将非叶子结点变为叶子节点,然后根据上述的方式同理计算就行
- 由于题目的二叉树是以数组的形式给出,并且题目已经说明是满二叉树,根据满二叉树的特性可以知道:节点i和节点i+1的父节点是节点i/2。因此,可以直接逆序遍历就可以得出结果
时间复杂度:O(n)
空间复杂度:O(1)
public int minIncrements(int n, int[] cost) {
int res=0;
for(int i=n-2;i>0;i-=2){
//取两个叶节点的差值,也就是需要操作的次数
res+=Math.abs(cost[i]-cost[i+1]);
// 将叶节点虚拟移除,即将叶节点值大的加入到父节点,使得父节点成为新的虚拟叶节点
cost[i/2]+=Math.max(cost[i],cost[i+1]);
}
return res;
}
有任何问题,欢迎评论区交流,欢迎评论区提供其它解题思路(代码),也可以点个赞支持一下作者哈😄~