C++ / ACM 自学笔记 · v5.0

P1090 合并果子

原题入口

对应章节:一、C++ 语法基础 → (十五) STL:容器、迭代器与算法;五、数据结构与算法 → (十一) 贪心算法。

(一) 任务与输出目标

每次合并两堆的花费为重量之和,求全部合并的最小花费。

(二) 建模与推导

每片果子的重量会在从叶子到最终堆的每次合并中计费,总代价是重量×叶深的总和。在一棵最优合并树里,可以把最小的两个重量换到最深的一对兄弟叶子,总代价不会增加。把它们合并成一个新重量后,剩下的问题形式相同,因此反复合并最小两堆。小根堆负责选取。

(三) 手算与过程

1,2,4:先合 1+2 花 3,再合 3+4 花 7,总计 10;先合 2+4 则花 6+7=13。

按具体数据展开的处理与返回过程

(四) 带注释实现

完整源文件

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;

void solve() {
    int n;
    cin >> n;
    priority_queue<int, vector<int>, greater<int>> heap;
    while (n--) {
        int x;
        cin >> x;
        heap.push(x);
    }
    int answer = 0;
    while (heap.size() > 1) {
        int a = heap.top();
        heap.pop();
        int b = heap.top();
        heap.pop();
        answer += a + b;
        heap.push(a + b); // 新堆将参加后续合并。
    }
    cout << answer << '\n';
}

signed main() {
    ios::sync_with_stdio(false); // 关闭同步,使用 cin 与 cout 完成输入输出。
    cin.tie(0), cout.tie(0);

    int T = 1; // 本题读入一组数据。
    // cin >> T; // 题目给出测试组数时开启,并在 solve 中处理一组。
    while (T--)
        solve();
}

(五) 复杂度

时间 \(O(n\log n)\),空间 \(O(n)\)。

返回本章题解索引 · 返回全部题号索引

查阅笔记正文与全部题解 · Ctrl / ⌘ K