给你一个整数 n
表示一棵 满二叉树 里面节点的数目,节点编号从 1
到 n
。根节点编号为 1
,树中每个非叶子节点 i
都有两个孩子,分别是左孩子 2 * i
和右孩子 2 * i + 1
。
树中每个节点都有一个值,用下标从 0 开始、长度为 n
的整数数组 cost
表示,其中 cost[i]
是第 i + 1
个节点的值。每次操作,你可以将树中 任意 节点的值 增加 1
。你可以执行操作 任意 次。
你的目标是让根到每一个 叶子结点 的路径值相等。请你返回 最少 需要执行增加操作多少次。
注意:
- 满二叉树 指的是一棵树,它满足树中除了叶子节点外每个节点都恰好有 2 个子节点,且所有叶子节点距离根节点距离相同。
- 路径值 指的是路径上所有节点的值之和
解题思路:
首先每个叶子节点相同的双亲节点的路径相同,所以首先把相同双亲节点的叶子节点的间距计算出来,则其为叶子节点的相加,为了使得相加的值为最小值,则需要加叶子节点以上的节点值。
所以所有叶子节点的上一个节点值加上叶子节点(最大的那个),这是把叶子节点的上一个节点当成叶子节点,一直运算到根节点的下一层即可。
C语言代码如下:
int minIncrements(int n, int* cost, int costSize)
{
int ans=0;
for(int i=n-2;i>0;i=i-2)
{ //首先把叶子节点的绝对值加上(因为其双亲节点一致,
所以这样可以确保每个双亲节点的路径长度一致)
ans+=abs(cost[i]-cost[i+1]);//计算同一双亲的叶子节点的差距的绝对值
//计算叶节点i和i+1的双亲节点的下标为i/2(整数除法)
//把双亲节点加上叶子节点最大值,则此时双亲节点变成叶子节点可以进行上诉操作
cost[i/2]+=fmax(cost[i],cost[i+1]);
}
return ans;
}