UVA 11997 K Smallest Sums

题目链接： https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1136

分析见《算法竞赛入门经典-训练指南》P190

#include <iostream>
#include <algorithm>
#include <map>
#include <vector>
#include <functional>
#include <string>
#include <cstring>
#include <queue>
#include <set>
#include <cmath>
#include <cstdio>


using namespace std;
#define IOS ios_base::sync_with_stdio(false)
typedef long long LL;
const int INF = 0x3f3f3f3f;
const double PI=4.0*atan(1.0);

const int maxn=800;
typedef struct Item{
    int s,b;
    Item(int s,int b):s(s),b(b){}
    bool operator<(const Item &a) const{
        return s>a.s;
    }
}I;
int k,a[maxn],b[maxn];
void merge(int *a,int *b,int *c)
{
    sort(a,a+k);
    sort(b,b+k);
    priority_queue<I> que;
    for(int i=0;i<k;i++)
        que.push(I(a[i]+b[0],0));
    for(int i=0;i<k;i++){
        I it=que.top(); que.pop();
        c[i]=it.s;
        if(it.b<k-1) que.push(I(it.s-b[it.b]+b[it.b+1],it.b+1));
    }
}
int main()
{
    while(scanf("%d",&k)!=EOF){
        for(int i=0;i<k;i++) scanf("%d",&a[i]);
        for(int i=0;i<k-1;i++){
            for(int j=0;j<k;j++) scanf("%d",&b[j]);
            merge(a,b,a);
        }
        for(int i=0;i<k-1;i++) printf("%d ",a[i]);
        printf("%d
",a[k-1]);
    }
}