http://acm.hdu.edu.cn/showproblem.php?pid=4546
思路:建立一个结构体记录一下当前总和、下一总和、当前位置;然后在进行加的时候要有两种,一种是跳过这个数加,另一种是不跳过加,这样就不会漏掉了。
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <algorithm>
using namespace std;
int a[10005];
struct node
{
int sum,nextsum,num;
friend bool operator<(node a,node b)
{
return a.nextsum>b.nextsum;
}
};
int main()
{
int T;
cin>>T;
int m,n;
node c,d;
int z=1;
while(T--)
{
cin>>m>>n;
int i;
priority_queue<node>Q;
for(i=0;i<m;i++)
{
cin>>a[i];
}
sort(a,a+m);
c.sum=0;
c.nextsum=a[0];
c.num=0;
Q.push(c);
int k=0;
while(!Q.empty())
{
d=Q.top();
Q.pop();
if(d.num>=m)continue;
c.sum=d.sum;
c.nextsum=d.sum+a[d.num+1];
c.num=d.num+1;
Q.push(c);
c.sum=d.nextsum;
c.nextsum=d.nextsum+a[d.num+1];
c.num=d.num+1;
Q.push(c);
k++;
if(k==n)
{
printf("Case #%d: %d\n",z++,d.nextsum);
break;
}
}
}
return 0;
}