题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4546
用优先队列维护数据,只需计算m次即可。
那么如何保证每次得到最小值。
先从小到大排序
例如当前和是sum,参与加和的最后一个数的下标是i;
那么最小值势必是sum+a[i+1]和sum+a[i+1]-a[i]中的一个。
那么就可以将sum+a[i+1]和sum+a[i+1]-a[i]都放进优先队列维护,每次取出队首,拓展两个值,进队,第m次出队的势必是第m小的数
下面是AC代码:
<pre name="code" class="cpp">#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=10005;
int n,m,a[maxn];
struct node
{
int s,p;
friend bool operator <(node a,node b)
{
return a.s>b.s;
}
};
int solve()
{
priority_queue<node>q;
node t;
a[n]=1<<30;//下标为i拓展的数不可能成为最小值
t.s=a[0];t.p=0;
q.push(t);
int sum,pos;
for(int i=1;i<=m;i++)
{
//cout<<"yes"<<endl;
t=q.top();q.pop();
sum=t.s;
pos=t.p;
t.s=sum+a[pos+1];
t.p=pos+1;
q.push(t);
t.s=sum+a[pos+1]-a[pos];
q.push(t);
}
return sum;
}
int main()
{
int t,Case=1;
cin>>t;
while(t--)
{
cin>>n>>m;
for(int i=0;i<n;i++) cin>>a[i];
sort(a,a+n);
cout<<"Case #"<<Case++<<": "<<solve()<<endl;
}
return 0;
}