给你一个整数 n 表示一棵 满二叉树 里面节点的数目,节点编号从 1 到 n 。根节点编号为 1 ,树中每个非叶子节点 i 都有两个孩子,分别是左孩子 2 * i 和右孩子 2 * i + 1 。
树中每个节点都有一个值,用下标从 0 开始、长度为 n 的整数数组 cost 表示,其中 cost[i] 是第 i + 1 个节点的值。每次操作,你可以将树中 任意 节点的值 增加 1 。你可以执行操作 任意 次。
你的目标是让根到每一个 叶子结点 的路径值相等。请你返回 最少 需要执行增加操作多少次。
注意:
满二叉树 指的是一棵树,它满足树中除了叶子节点外每个节点都恰好有 2 个子节点,且所有叶子节点距离根节点距离相同。
路径值 指的是路径上所有节点的值之和。
示例 1:
输入:n = 7, cost = [1,5,2,2,3,3,1]
输出:6
解释:我们执行以下的增加操作:
- 将节点 4 的值增加一次。
- 将节点 3 的值增加三次。
- 将节点 7 的值增加两次。
从根到叶子的每一条路径值都为 9 。
总共增加次数为 1 + 3 + 2 = 6 。
这是最小的答案。
示例 2:
输入:n = 3, cost = [5,3,3]
输出:0
解释:两条路径已经有相等的路径值,所以不需要执行任何增加操作。
class Solution {
/*先从叶子节点开始,与兄弟节点相比较
假设两个兄弟节点x,y,且x<y,他们的上层路径相同,
那么要让两个节点路径值之和相等,要让x+|x-y|
*/
public:
int minIncrements(int n, vector<int>& cost) {
int ans=0;
for(int i=n-2;i > 0;i-=2)
{
ans+=abs(cost[i] - cost[i+1]);
//查看以此为根的两个子树的值是否相等
//孩子节点已经调整完成,只能调整子树的根节点
cost[i/2]+=max(cost[i],cost[i+1]);
}
return ans;
}
};
https://leetcode.cn/problems/make-costs-of-paths-equal-in-a-binary-tree/submissions/505496710/