AcWing146. 序列(贪心)

题干：

给定m个序列，每个包含n个非负整数。

现在我们可以从每个序列中选择一个数字以形成具有m个整数的序列。

很明显，我们一共可以得到$n^m$个这种序列， 然后我们可以计算每个序列中的数字之和，并得到$n^m$个值。

现在请你求出这些序列和之中最小的n个值。

对于每组测试用例，均以递增顺序输出最小的n个序列和，数值之间用空格隔开。

0<m≤1000
0<n≤2000

思路：

首先我们来看样例，输入为
1
2 3
1 2 3
2 2 3
输出为
3 3 4
根据题意我们可以得出所有的序列和为：1+2 1+2 1+3 2+2 2+2 2+3 3+2 3+2 3+3这九个数，则最小的n个为3 3 4。
如果我们将m(设m=2)个序列先从小到大排序，这样我们要求的系列和就变成了
A0+B0 A1+B0 … An-1+B0
A0+B1 A1+B1 … $An-1$+B1
…
A0+Bn-1 A1+Bn-1 … An-1+Bn-1
由于我们的系列是有序的，则得到的序列和也是有序的，每组序列和的第一个一定是该组中最小的，所以我们就从这些小的中选一个最小的，选完之后去掉，则该组中最小的变成了Ai+1 + Bj；这样重复n次，我们就得到了需要的最小的n个序列和。

然后考虑m>2的情况，我们可以先合并前两个，得到一个长度为n的序列和，再依次和后面的每一个进行合并，这样对于每一次操作都相当于m=2的情况。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <iostream>
#include <algorithm>
#define ll unsigned long long
using namespace std;
int a[4010],b[4010],c[4010],n,m;
pair<int,int> t;
priority_queue<pair<int,int>, vector<pair<int,int> >, greater<pair<int,int> >>q;
int main()
{
    int t1;
    scanf("%d",&t1);
    while(t1--){
        scanf("%d%d",&m,&n);
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        sort(a,a+n);
        for(int j=1;j<m;j++){
            while(!q.empty()){
                q.pop();
            }
            for(int i=0;i<n;i++)
            	scanf("%d",&b[i]);
            sort(b,b+n);
            for(int i=0;i<n;i++)
            	q.push({a[0]+b[i],0});
            for(int i=0;i<n;i++){
            	t=q.top();
            	q.pop();
            	c[i]=t.first;
            	q.push({t.first+a[t.second+1]-a[t.second],t.second+1});
            	//移动a数组的下标，t.first中已经在上次加上了a[t.second]，所以注意要减去
            }
            memcpy(a,c,n*4);//int占4个字符
        }
        for(int i=0;i<n;i++)
        	printf("%d ",a[i]);
        printf("\n");
    }
    return 0;
}