天梯选拔赛A套题中一道25分的题目。我本人比较喜欢题目很短,和数据有关的题目,因此当时一心想做出来这道题目,可以说执念很深了,结果看了四十分钟楞是没有什么头绪。赛后,看了答案,发现是用哈夫曼树+优先队列做,代码很简单,思绪就是哈夫曼树,没有学过的知识点就不说什么了,优先队列的知识点有接触过,还是刷题时没好好刷,都忘了。现在借这道题来好好复习一下优先队列吧。先把这道题的代码贴上来:
#include <cmath>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <iomanip>
#include <cstring>
#include <map>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
typedef long long ll;
const int maxn = 20000 + 10;
int l[maxn],t;
void solve()
{
ll sum = 0;
while(t>1)
{
int mmin1 = 0,mmin2 = 1;
if(l[mmin1]>l[mmin2])
swap(mmin1,mmin2);
for(int i=2;i<t;i++)
{
if(l[i]<l[mmin1])
{
mmin1 = i;
mmin2 = mmin1;
}
else if(l[i]<mmin2)
mmin2 = i;
}
int mmin = l[mmin1] + l[mmin2];
sum += mmin;
if(mmin1 == t-1)
swap(mmin1,mmin2);
l[mmin1] = mmin;
l[mmin2] = l[t-1];
t--;
}
printf("%lld\n",sum);
}
int main()
{
scanf("%d",&t);
for(int i=0; i<t; i++)
cin>>l[i];
solve();
return 0;
}
这是用优先队列做的,代码简单,效率高
#include <cmath>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <iomanip>
#include <cstring>
#include <map>
#include <vector>
#include <queue>
#include <stack>
using namespace std;
const int maxn = 20000+10;
typedef long long ll;
int l[maxn],n;
void solve()
{
ll sum = 0;
priority_queue<int,vector<int>,greater<int> >q;
for(int i=0; i<n; i++)
q.push(l[i]);
while(q.size()>1)
{
int l1,l2;
l1 = q.top();
q.pop();
l2 = q.top();
q.pop();
int mmin = l1 + l2;
sum += mmin;
q.push(mmin);
}
printf("%lld\n",sum);
}
int main()
{
scanf("%d",&n);
for(int i=0; i<n; i++)
scanf("%d",&l[i]);
solve();
return 0;
}
好吧,我实在是太懒了,附上个链接吧,想学习的可以自己去看。 点击打开链接