一、题目概述
给你一个整数n表示一棵满二叉树里面节点的数目,节点编号从1到n。根节点编号为1,树中每个非叶子节点i都有两个孩子,分别是左孩子2*i和右孩子2*i+1。树中每个节点都有一个值,用下标从0开始、长度为n的整数数组cost表示,其中cost[i]是第i+1个节点的值。每次操作,你可以将树中任意节点的值增加1。你可以执行操作任意次。你的目标是让根到每一个叶子结点的路径值相等。请你返回最少需要执行增加操作多少次。
- 满二叉树:指的是一棵树,它满足树中除了叶子节点外每个节点都恰好有 2 个子节点,且所有叶子节点距离根节点距离相同。
路径值:指的是路径上所有节点的值之和。
输入输出示例:
输入:n = 7, cost = [1,5,2,2,3,3,1]
输出:6
输入:n = 3, cost = [5,3,3]
输出:0
二、思路
我们首先来看,对于任一叶子结点,它的值为x,它的兄弟结点的值为y。对于树上其他的结点,它们要么同时是这两个叶节点的祖先,要么同时不是这两个叶节点的祖先。对这些节点进行一次操作,要么同时增加了根到这两个叶节点的路径值 1,要么没有任何效果。因此,要想使得根到这两个叶节点的路径值相等,我们只能增加 x和 y本身。
因此如果二者的路径值不相等,就一定是这两个叶子节点的值不等导致的,由于我们只能增加值,且每次+1,所有我们应该让较小的叶子节点值变成较大的叶子节点值,增加的操作数就是二者之差,也就是进行|x-y|次操作。
如下图所示,叶子节点4和5到根节点的路径,除了两者之外,其余节点都相等,所以为了使得两者路径值相同,可以将节点4的值变成3即可。
经过一次处理后,每两个叶子节点一组,它们之间到根节点的路径值就相同了。所以我们只需要再看他们的父节点即可,也就是这些父节点的的父节点为根节点的子树要实现同样的条件。如下图,可以把节点2和节点3这一层看成新的叶子节点,值为对应的路径值,所以这一层操作就是将节点3的路径值5变成节点2的路径值8,操作了三次。
所以自底向上进行处理,先使得叶子节点到最近的根节点的子树的路径值相同,再一层一层往上,扩充子树,使得更多叶子节点到更顶层的根节点的路径值相同,直到到达根节点。因此,在整个自底向上的过程中,只要统计每个非叶子节点的两个子节点当前路径值之差,即可以得到最终的结果。由于树由数组形式给出,叶子节点在末尾出现,所以我们对数组进行逆序遍历;又因为是满二叉树,所以连续两个节点肯定是兄弟节点,所以我们每次遍历步长为2;对于满二叉树,父节点和左右孩子节点的关系为:.当i/2取取整>=1时,i/2取整单元是其父结点。
三、复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)
四、代码
#include <iostream>
#include <vector>
using namespace std;
class Solution
{
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;
}
};
int main()
{
int n = 7;
vector<int> cost = {1, 5, 2, 2, 3, 3, 1};
int ans = Solution().minIncrements(n, cost);
cout << ans << endl;
n = 3;
cost = {5, 3, 3};
ans = Solution().minIncrements(n, cost);
cout << ans << endl;
}
OK,问题解决,求点赞关注!如果有其他思路,或者其他语言的解题方式,欢迎大家在下面热烈讨论~