P3765k个最小和

问题描述

有k个整数数组，各包含k个元素，从每个数组中选取一个元素加起来，可以得到k^k个和，求这些和中最小的k个值。

输入格式

第一行，一个整数k(k<=500)
接下来k行，每行k个正整数(<=1000000)

输出格式

一行，k个有小到大排列的整数，表示最小的k个和

样例输入 1

3
1 8 5
9 2 5
10 7 6

样例输出 1

9 10 12

样例输入 2

2
1 1
1 2

样例输出 2

2 2

题解

多路合并水题
显然之前所得k个最小和加上此时数组的最小和为当前最小和。
排序后能得到

1.1>1.2>1.3…
2.1>2.2>2.3..
3.1>3.2>3.3..
k.1>k.2>k.3..
所以先将1-k进队后依次讨论用优先队列维护

代码

#include<stdio.h>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
#include<iostream>
using namespace std;
#define inf 1e9
#define pr cout<<
priority_queue<int>q;
int temp[505],a[505];
int k,i,j;
void work(){
    for(i=1;i<=k;i++){
        q.push(a[1]+temp[i]);
    }
    int cnt=0;
    for(i=1;i<=k;i++)
    {
        if(cnt==k) break;
        cnt++;

        for(j=2;j<=k;j++)
        {
            if(a[j]+temp[i]<q.top()){
                q.push(a[j]+temp[i]);
                q.pop();
                cnt++;
            }
            else break;
        }
    }
    for(i=k;i>=1;i--) a[i]=q.top(),q.pop();

}
int main()
{
    int i,j,n;
    scanf("%d",&k);
    for(i=1;i<=k;i++) scanf("%d",&a[i]);
    sort(a+1,a+k+1);
    for(i=2;i<=k;i++){
        for(j=1;j<=k;j++) scanf("%d",&temp[j]);
        sort(temp+1,temp+k+1);
        work();
    }
    for(i=1;i<=k;i++) cout<<a[i]<<" ";
}