UVA 11997 【K Smallest Sums】

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
struct  Item
{
  int s,b;
  Item(int s,int b):s(s),b(b){}
  bool operator <(const Item &a) const{
      return s>a.s;
  }
} ;
void merge(int *A,int *B,int *C,int n)
{
    priority_queue<Item>q;
    for(int i=0;i<n;i++)
    q.push(Item(A[i]+B[0],0));
    for(int i=0;i<n;i++)
    {
     Item item =q.top();q.pop();
     int b=item.b;
     C[i]=item.s;
     if(b+1<n)  q.push(Item(item.s-B[b]+B[b+1],b+1));//细细体会此处 
    }
}
const int maxn=768;
int A[maxn][maxn];
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            scanf("%d",&A[i][j]);
            sort(A[i],A[i]+n);
        }
        for(int i=1;i<n;i++)
        {
            merge(A[0],A[i],A[0],n);
        }
        printf("%d",A[0][0]]);
        for(int i=1;i<n;i++)
        printf(" %d",A[0][i]);
        printf("\n");    
    }
    return 0;
}