uva11997 K Smallest Sums&&UVALive 3135 Argus（优先队列，多路归并）

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<cctype>
#include<sstream>
using namespace std;
#define pii pair<int,int>
#define LL long long int
const double eps=1e-;
const int INF=;
const int maxn=+;
int n,a[maxn][maxn];
struct node
{
    int s,b;
    node(int s,int b): s(s),b(b){}
    bool operator < (const node &x) const
    {
        return s>x.s;
    }
};
priority_queue<node>pq;
void he(int *A,int *B,int *C)
{
    while(!pq.empty()) pq.pop();
    for(int i=; i<n; i++)
    {
        pq.push(node(A[i]+B[],));
    }
    for(int i=; i<n; i++)
    {
        node t=pq.top();
        pq.pop();
        C[i]=t.s;
        pq.push(node(t.s-B[t.b]+B[t.b+],t.b+));
    }
}
int main()
{
    //freopen("in1.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(scanf("%d",&n)==)
    {
        for(int i=; i<n; i++)
        {
            for(int j=; j<n; j++)
            {
                scanf("%d",&a[i][j]);
            }
            sort(a[i],a[i]+n);
        }
        for(int i=; i<n; i++)
        {
            he(a[],a[i],a[]);
        }
        for(int i=; i<n-; i++) printf("%d ",a[][i]);
        printf("%d\n",a[][n-]);
    }
    return ;
}

uva11997

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<cctype>
#include<sstream>
using namespace std;
#define pii pair<int,int>
#define LL long long int
const double eps=1e-;
const int INF=;
const int maxn=+;
int k,a,b,ans[+];
string s;
struct node
{
    int num,t,p;
    node (int num,int t,int p):num(num),t(t),p(p){}
    bool operator <(const node &x)const
    {
        if(t!=x.t)
        {
            return t>x.t;
        }
        else
        {
            return num>x.num;
        }
    }
};
priority_queue<node>pq;
int main()
{
    //freopen("in1.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(cin>>s)
    {
        if(s[]=='#')
        {
            scanf("%d",&k);
            for(int i=;i<k;i++)
            {
                node tmp=pq.top();
                pq.pop();
                ans[i]=tmp.num;
                pq.push(node(tmp.num,tmp.t+tmp.p,tmp.p));
            }
        }
        else
        {
            scanf("%d%d",&a,&b);
            pq.push(node(a,b,b));
        }
        for(int i=;i<k;i++)
        {
            printf("%d\n",ans[i]);
        }
    }
    return ;
}

uvalive 3135