[uva11997]k个最小和

一个k*k的矩阵，每行选取一个数相加则得到一个和，求最小的前k个和。

k<=750

已知前m行最小的前k个和d[1]…d[k]，则前m+1行最小的前k个和都必定是d[i](i<=k)+a[m+1][x]。排序，枚举x，用优先队列处理。

学会了个小技巧：

node形式的优先队列，想直接插入元素组成node

struct node里加一句node (int sum,int b):sum(sum),b(b) {} 实际调用：q.push(node(sum,b));

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<vector>
 #include<map>
 #include<queue>
 using namespace std;

 const int K=;
 int k,a[K][K];
 bool cmp(int x,int y){return x<y;}

 struct node{
     int sum,b;
     node (int sum,int b):sum(sum),b(b) {}
     bool operator < (const node &x) const {
         return sum > x.sum;
     }
 };

 priority_queue<node> q;

 void my_merge(int *A,int *B)
 {
     while(!q.empty()) q.pop();
     for(int i=;i<=k;i++) q.push(node(A[i]+B[],));
     for(int i=;i<=k;i++)
     {
         node x=q.top();q.pop();
         A[i]=x.sum;
         if(x.b<=k-) q.push(node(x.sum-B[x.b]+B[x.b+],x.b+));
     }
 }

 int main()
 {
     // freopen("a.in","r",stdin);
     // freopen("a.out","w",stdout);

     while(scanf("%d",&k)!=EOF)
     {
         for(int i=;i<=k;i++)
             for(int j=;j<=k;j++)
                 scanf("%d",&a[i][j]);

         sort(a[]+,a[]+k+,cmp);
         for(int i=;i<k;i++)
         {
             sort(a[i+]+,a[i+]+k+,cmp);
             my_merge(a[],a[i+]);
         }
         printf("%d",a[][]);
         for(int i=;i<=k;i++) printf(" %d",a[][i]);printf("\n");
     }

     return ;
 }