堆：序列

题目链接：https://www.acwing.com/problem/content/148/
题意：给定m个序列，每个包含n个非负整数。现在我们可以从每个序列中选择一个数字以形成具有m个整数的序列。
很明显，我们一共可以得到n^m个这种序列， 然后我们可以计算每个序列中的数字之和，并得到n^m个值。现在请你求出这些序列和之中最小的n个值。
数据范围
0<m≤1000,
0<n≤2000
输入样例：
1
2 3
1 2 3
2 2 3
输出样例：
3 3 4
思路：要合并m个序列不好处理，那我们就每次合并两个序列，那么就只需要合并m - 1次了。而最核心的地方就是两个序列合并后保留前n个序列和，这里采用分组法，例如：
a1 a2 a3 …… an
b1 b2 b3 …… bn
先把a组从小到大排序，然后分成n组：
a1+b1 , a1+b2 , a1+b3 , …… a1+bn
a2+b1 , a2+b2 , a2+b3 , …… a2+bn
…………………………………………
an+b1 , an+b2 , an+b3 , …… an+bn
这样最 每组就是升序，最左列的数就是Min,在最左列找到一个最小值ai+b1加入合并集合中，那么原本ai+b1的位置就被ai+b2替代，再如此循环寻找到n个min后，就得到了两个序列的前n小个和了，时间复杂度也才O(n*logn)，完全可以了。
代码实现：

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>

using namespace std;
typedef pair<int,int> PII;
const int N = 2010;

int n, m;
int a[N], b[N], c[N];

void solve()
{
    //优先队列定义一个小根堆，第一键值存和，第二键值存下标
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    for (int i = 0; i < n; i ++ ) heap.push({a[0] + b[i], 0});
    for (int i = 0; i < n; i ++ ){
        auto t = heap.top();
        heap.pop();
        c[i] = t.first;
        //加入的新元素和上一个元素差别就是a[]的下标加一
        heap.push({t.first + a[t.second + 1] - a[t.second], t.second + 1});
    }
    //再将c的值赋值回a中
    memcpy(a, c, 4 * n);
}

int main()
{
    int T;
    cin >> T;
    while (T -- ){
        cin >> m >> n;
        for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
        sort(a, a + n);
        //m - 1次合并操作
        for (int i = 1; i < m; i ++ ){
            for (int j = 0; j < n; j ++ ) scanf("%d", &b[j]);
            //sort(b, b + n);b可以不用再排序
            solve();
        }
        
        for (int i = 0; i < n; i ++ ) printf("%d ", a[i]);
        cout << endl; 
    }
    return 0;
}