贪心+深搜，LeetCode 2673. 使二叉树所有路径值相等的最小代价

一、题目

1、题目描述

给你一个整数 n 表示一棵 满二叉树 里面节点的数目，节点编号从 1 到 n 。根节点编号为 1 ，树中每个非叶子节点 i 都有两个孩子，分别是左孩子 2 * i 和右孩子 2 * i + 1 。

树中每个节点都有一个值，用下标从 0 开始、长度为 n 的整数数组 cost 表示，其中 cost[i] 是第 i + 1 个节点的值。每次操作，你可以将树中 任意 节点的值 增加 1 。你可以执行操作 任意 次。

你的目标是让根到每一个 叶子结点 的路径值相等。请你返回 最少 需要执行增加操作多少次。

注意：

• 满二叉树 指的是一棵树，它满足树中除了叶子节点外每个节点都恰好有 2 个子节点，且所有叶子节点距离根节点距离相同。
• 路径值 指的是路径上所有节点的值之和。

2、接口描述

​

class Solution {
public:
    int minIncrements(int n, vector<int>& cost) {
        
    }
};

3、原题链接

2673. 使二叉树所有路径值相等的最小代价

---

二、解题报告

1、思路分析

正难则反

对于最终的结果，如果去掉根节点，那么剩下两颗子树各自从根到叶子节点的路径值都是相同的

由于操作只能增加节点的值，所以我们对于一个节点而言，将两个子节点所在子树调整完后的最大路径值分别为l，r，那么对于这个节点的操作次数就是abs(r-l)

这个节点到达叶子节点的最大路径值就是max(l, r) + cost[x]

2、复杂度

时间复杂度： O(n)空间复杂度：O(logn)

3、代码详解

​

class Solution {
public:
    int minIncrements(int n, vector<int>& cost) {
        int ret = 0;
        function<int(int)> dfs = [&](int x){
            int lc = x << 1, rc = x << 1 | 1;
            if(lc > n) return cost[x - 1];
            int l = dfs(lc), r = dfs(rc);
            ret += abs(r - l);
            return max(l, r) + cost[x - 1];
        };
        dfs(1);
        return ret;
    }
};