uva 11997 K Smallest Sums

解题思路：STL —— priority_queue

///////////////////////////////////////////////////////////////////////////
//problem_id: uva 11997
//user_id: SCNU20102200088
///////////////////////////////////////////////////////////////////////////

#include <algorithm>
#include <iostream>
#include <iterator>
#include <iomanip>
#include <cstring>
#include <cstdlib>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <set>
#include <map>
using namespace std;

///////////////////////////////////////////////////////////////////////////
#pragma comment(linker,"/STACK:1024000000,1024000000")

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
///////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////
const double EPS=1e-9;
const double PI=acos(-1.0);
const double E=2.7182818284590452353602874713526;

const int x4[]={-1,0,1,0};
const int y4[]={0,1,0,-1};
const int x8[]={-1,-1,0,1,1,1,0,-1};
const int y8[]={0,1,1,1,0,-1,-1,-1};
///////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////
typedef long long LL;

typedef int T;
T max(T a,T b){ return a>b? a:b; }
T min(T a,T b){ return a<b? a:b; }
T gcd(T a,T b){ return b==0? a:gcd(b,a%b); }
T lcm(T a,T b){ return a/gcd(a,b)*b; }
///////////////////////////////////////////////////////////////////////////

///////////////////////////////////////////////////////////////////////////
//Add Code:
int k,a[755][755];

struct Node{
    int x,y;
    bool operator <(const Node &a) const{
        return x>a.x;
    }
};

void merge(int *x,int *y,int *z){
    priority_queue<Node> pq;
    int i,j;
    for(i=0;i<k;i++){
        Node res=Node{y[i]+z[0],0};
        pq.push(res);
    }
    for(i=0;i<k;i++){
        Node now=pq.top();
        pq.pop();
        x[i]=now.x;
        j=now.y;
        if(j+1<k){
            Node res=Node{x[i]-z[j]+z[j+1],j+1};
            pq.push(res);
        }
    }
}
///////////////////////////////////////////////////////////////////////////

int main(){
    ///////////////////////////////////////////////////////////////////////
    //Add Code:
    int i,j,ans[755];
    while(scanf("%d",&k)!=EOF){
        for(i=0;i<k;i++){
            for(j=0;j<k;j++) scanf("%d",&a[i][j]);
            sort(a[i],a[i]+k);
        }
        merge(ans,a[0],a[1]);
        for(i=2;i<k;i++) merge(ans,ans,a[i]);
        printf("%d",ans[0]);
        for(i=1;i<k;i++) printf(" %d",ans[i]);
        printf("
");
    }
    ///////////////////////////////////////////////////////////////////////
    return 0;
}

///////////////////////////////////////////////////////////////////////////
/*
Testcase:
Input:
3
1 8 5
9 2 5
10 7 6
2
1 1
1 2
Output:
9 10 12
2 2
*/
///////////////////////////////////////////////////////////////////////////