多元Huffman编码问题Time Limit: 1000 ms Memory Limit: 65536 KiB Problem Description 在一个操场的四周摆放着n堆石子。现要将石子有次序地合并成一堆。规定每次至少选2 堆最多选k堆石子合并成新的一堆,合并的费用为新的一堆的石子数。试设计一个算法,计算出将n堆石子合并成一堆的最大总费用和最小总费用。 Input 输入数据的第1 行有2 个正整数n和k(n≤100000,k≤10000),表示有n堆石子,每次至少选2 堆最多选k堆石子合并。第2 行有n个数(每个数均不超过 100),分别表示每堆石子的个数。 Output 将计算出的最大总费用和最小总费用输出,两个整数之间用空格分开。 Sample Input Sample Output Hint 请注意数据范围是否可能爆 int。 Source |
最小总费用 :
样例:
7 5
5 9 12 13 16 22 45
那么 最小值应该是148 ,但是如果按照我们的贪心算法是先将最小的情况合并的话 那么就会出错
过程:
- 1:先把最小的五个加起来
- 5 9 12 13 16=55
- 2:形成新的系列:55 22 45
- 3:然后 55 + 22 + 45 = 122+55 = 177 那就错了
改进:
- 5 9 12 先合并为26
- 然后13 16 22 26 45 再合并就122
- 结果为122+26 = 148 over
问题来了,那么要怎么解决?
发现往前面+0即可 就是
样例:
9 5
0 0 5 9 12 13 16 22 45
- 那么就是0 0 5 9 12 = 26
- 然后13 16 22 26 45----->122 + 26 = 148
- 附加:
- 你看我们是不是都凑够了最大的合并数(5),因为要求最小总费用,所以要求合并次数尽量少,
- 经过优先队列的排序后,我们只要每次合并最大的合并数(5),就可以达到最小。over)
- 其实要加的0的个数和n和k之间是有一个关系的不过我还没想出来。。。
最大总费用:
至于最大值就是最大的两个数合并再扔进去即可~~~
- 附加:
- 为要求最大总费用,所以要求合并次数尽量多,
- 在经过优先队列的排序后,每次选取最大的两个合并就over~~
#include <iostream>
#include<stdio.h>
#include<queue>
using namespace std;
typedef long long ll;
int main()
{
int n,k;
scanf("%d%d",&n,&k);
priority_queue<int >p;//降序队列
priority_queue<int,vector<int >,greater<int > >q;//升序队列
for(int i=0; i<n; i++)
{
int x;
scanf("%d",&x);
q.push(x);
p.push(x);
}
ll ans1,ans2,temp;
ans1=ans2=temp=0;
while(q.size()%(k-1)!=1)
{
q.push(0);
}
while(q.size()>1)//最小
{
for(int i=0; i<k; i++)
{
int x=q.top();
q.pop();
temp+=x;
}
ans1+=temp;
q.push(temp);
temp=0;
}
temp=0;
while(p.size()>1)
{
int x=p.top();
p.pop();
int y=p.top();
p.pop();
temp=x+y;
p.push(temp);
ans2+=temp;
temp=0;
}
printf("%lld %lld\n",ans2,ans1);
return 0;
}