POJ2442 Sequence

题目链接：https://vjudge.net/problem/POJ-2442

题目大意：

　　有 \(m\) 个序列，每个序列 \(n\) 个数。在每个序列中选一个数，会选出 \(m\) 个数，对应一个和。那么对于所有的数，总共有 \(m^n\) 种选法，要求从小到大输出其中和最小的 \(n\) 种选法对应的和。

知识点：　　优先队列

解题思路：

　　对于 \(2\) 个序列，我们可以先把这两个序列从小到大排序，然后用第一个序列的第一项分别去加第二个序列的各项，得到 \(n\) 个和，放到一个优先队列（大顶堆）中；接下来用第一序列的第二项分别去加第二序列的各项，如果和小于优先队列的首项，那么首项出队，新的和入队，否则结束对第一序列第二项的操作，转而操作第一序列第三项；如此进行直到操作完第一序列的所有项，得到了第一第二序列各项相加的前 \(n\) 小的和。

　　那么对于 \(3\) 个序列的问题，我们可以先用上述方法合并前两个序列，得到一个 \(n\) 项的序列，再和第三序列合并出答案。至于大于 \(3\) 个序列的问题要怎么做，相信读者自己心里已经有答案了吧。

AC代码：

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int m,n;
int inp[105][2005];
priority_queue<int> q;
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&m,&n);
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++)
                scanf("%d",&inp[i][j]);
            sort(inp[i]+1,inp[i]+1+n);
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<=n;j++)   q.push(inp[i][1]+inp[i+1][j]);
            for(int j=2;j<=n;j++){
                for(int k=1;k<=n;k++){
                    if(inp[i][j]+inp[i+1][k]>=q.top())   break;
                    else{
                        q.pop();
                        q.push(inp[i][j]+inp[i+1][k]);
                    }
                }
            }
            int ind=n;
            while(!q.empty()){
                inp[i+1][ind]=q.top();
                q.pop();
                ind--;
            }
        }
        for(int i=1;i<=n;i++){
            if(i!=1)    printf(" ");
            printf("%d",inp[m][i]);
        }
        printf("\n");
    }
    return 0;
}