papamelon 217. 栅栏修理 Fence Repair(挑战程序设计竞赛)


地址 https://www.papamelon.com/problem/217

解答
将切割逆向看,可以看做是将多段木棒粘合起来。粘合的成本是两个木棒的长度

可以看到处于树最底层的木棒长度在每次粘合中都加入了进去。
所以底层的节点应该越短越好,比如5.
这就是一种贪心思想。具体原理可以参考哈夫曼树。
由于粘合后的木棒也需要作为木棒供下次粘合选择,所以不仅仅是一开始将所有起始的木棒排序就好了。
我们需要配合使用堆这个数据结构来进行元素排序。
每次取两段最短木棒粘合,然后将粘合的木棒放入选择堆中,按照长短排序。一共n次,排序时间为logn。 总体时间复杂度为 O(nlogn)

#include 
#include 
using namespace std;

using ll = long long;

priority_queue, greater> pq;

int main() {
  int n;
  cin >> n;
  for (int i = 0; i < n; i++) {
    int x;
    cin >> x;
    pq.push(x);
  }

  ll ret = 0;
  while (pq.size() > 1) {
    ll x = pq.top(); pq.pop();
    ll y = pq.top(); pq.pop();
    ret += x + y;
    pq.push(x + y);
  }
  cout << ret << endl;
  return 0;
}

我的视频题解空间