借鉴别人的思路,使用很简单的哈夫曼树。并且使用优先队列。
题目的意思是。要把一块木板切成所需要的n块,但是每切一刀需要收费。收费的方法是(网上有一些人理解有误):
每当在一块木板上动刀子,这块木板的原有长度代表了这刀的费用。
思路:
切割的过程可以用一棵二叉树表示,每个节点有两个或者没有子节点,
节点存储切割过程中每块木板的长度。父节点的值=两个子节点相加
根节点是原始木板长度。每块所需要的木板都是叶子节点。
非叶子节点的值之和就是要花的总价格。
定理1:
如果要使得总价格最少,就要求高层(靠近根)的每一个节点的值 大于 低层每一个节点的值
证明:
运用反证法,假如某一棵二叉树,总价格最低,且不满足定理1。
假如第N层a节点大于第M层b节点,N>M a-b = c
那么,我们可以构造出一棵二叉树,即把第N层的a节点和M层的b节点交换(如果是非叶子节点,交换子树),之后保证 父节点的值=两个子节点相加 。
a值的减少c会使得自己的每个先辈节点减少c, b值增加c也会使自己每个先辈节点增加c。由于a在b的下层,导致a的先辈比b的先辈多,
每个先辈节点必定不是叶子节点,这就使得总价格降低,和原二叉树总价最低矛盾。
定理2:
每一颗二叉树,只要高层(靠近根)的每一个节点的值 大于 低层每一个节点的值,那么这些二叉树所代表的总价值相同。
证明:
原理同定理1,通过同一层的节点(或者子树)相互交换,构造出的任意一棵满足要求的二叉树,
由于先辈节点值增加,减少的值相互抵消,所以总价值不变。
由于哈弗曼树满足我们所需要的二叉树的性质(这也同时说明,满足总费用最少的二叉树不一定是哈夫曼树),所以这道题用哈夫曼树来解。
#include<iostream>
#include<cstdio>
#include<queue>
#define inf 99999
using namespace std;
struct cmp {
bool operator()(int x, int y) {
return x > y;
}
};
int main()
{
int p;
priority_queue<int, vector<int>, cmp>q;
int t, n, m, M;
long long sum = 0;
scanf("%d", &t);
for (int i = 0; i < t; i++)
{
scanf("%d", &p);
q.push(p);
}
while (!q.empty())
{
n = q.top();
q.pop();
m = q.top();
q.pop();
M = n + m;
sum += M;
if (q.empty())
break;
q.push(M);
}
cout << sum << endl;
return 0;
}