[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=800;
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=1;i<=k;i++) q.push(node(A[i]+B[1],1));
    for(int i=1;i<=k;i++)
    {
        node x=q.top();q.pop();
        A[i]=x.sum;
        if(x.b<=k-1) q.push(node(x.sum-B[x.b]+B[x.b+1],x.b+1));
    }
}

int main()
{
    // freopen("a.in","r",stdin);
    // freopen("a.out","w",stdout);
    
    while(scanf("%d",&k)!=EOF)
    {
        for(int i=1;i<=k;i++)
            for(int j=1;j<=k;j++)
                scanf("%d",&a[i][j]);
                
        sort(a[1]+1,a[1]+k+1,cmp);
        for(int i=1;i<k;i++)
        {
            sort(a[i+1]+1,a[i+1]+k+1,cmp);
            my_merge(a[1],a[i+1]);
        }
        printf("%d",a[1][1]);
        for(int i=2;i<=k;i++) printf(" %d",a[1][i]);printf("\n");
    }
    
    return 0;
}