优先队列 - Sequence - POJ 2442

优先队列 - Sequence - POJ 2442

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

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

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

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

输入格式

第一行输入一个整数T，代表输入中包含测试用例的数量。

接下来输入T组测试用例。

对于每组测试用例，第一行输入两个整数m和n。

接下在m行输入m个整数序列，数列中的整数均不超过10000。

输出格式

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

每组输出占一行。

数据范围
$0<m\le 1000,0<n\le 2000$

输入样例：

1
2 3
1 2 3
2 2 3

输出样例：

3 3 4

分析：

$我们要从m个长度为n的序列中各挑选m个数得到新序列再求和，再从和的序列中挑选前n小的数，$

$等价于先求和，再从m个和的序列中挑选n个数组合成新的序列，输出新序列前n小的数。$

$而合并m个序列，可以两两合并m-1次，每次两两合并产生n^2个数，保留前n小的数再继续合并。$

$暴力合并共操作(m-1)n^2次，考虑优化合并方式。$

$$\begin{gathered} 考虑合并序列: \\ {a}_1,a_2,...,a_n和b_1,b_2,...,b_n \end{gathered}$$

$我们首先将序列a从小到大排序，接着将两个序列组合而成的n^2个数分组：$

$$\begin{gathered} 1)、a_1+b_1,a_2+b_1,...,a_n+b_1 \\ 2)、a_1+b_2,a_2+b_2,...,a_n+b_2 \\ ... \\ n)、a_1+b_n,a_2+b_n,...,a_n+b_n \end{gathered}$$

$因为序列a以按从小到大顺序排好，所以每一组的第一个数就是该组的最小值，我们先挑出n组的最小值：$

$$\begin{gathered} 1)、a_1+b_1 \\ 2)、a_1+b_2 \\ ... \\ k)、a_1+b_k \\ ... \\ n)、a_1+b_n \end{gathered}$$

$那么这n^2个数中的最小的数一定在这n个数中，我们假设a_1+b_k是最小的，1\le k\le n。$

$挑出a_1+b_k后，第k组的最小值就变成了a_2+b_k。$

$于是n组中的最小值变成了$

$$\begin{gathered} 1)、a_1+b_1 \\ 2)、a_1+b_2 \\ ... \\ k)、a_2+b_k \\ ... \\ n)、a_1+b_n \end{gathered}$$

$这样，继续从以上n个数中挑选最小的数，就得到了这n^2个数中的次小数。$

$以此类推，重复进行n次操作即可。$

$这个过程可以用小根堆来维护，时间复杂度O(nlogn)。$

$共合并m-1次，总的时间复杂度为O(mnlogn)。$

具体落实：

$小根堆以两个序列的和为第一关键字，由于挑选某个数后，要将其后的数加入堆中，因此还需存储对应的下标。$

$用pair来存储。$

代码：

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

#define P pair<int,int>
#define x first
#define y second

using namespace std;

const int N=2010;

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

void merge()   //将序列a和b组合后挑出前n小的数存储到a中
{
    priority_queue<P,vector<P>,greater<P> > heap;
    for(int i=0;i<m;i++) heap.push({a[0]+b[i],0});
    
    for(int i=0;i<m;i++)
    {
        P t=heap.top();
        heap.pop();
        int s=t.x,p=t.y;
        c[i]=s;
        heap.push({s-a[p]+a[p+1],p+1});
    }
    
    for(int i=0;i<m;i++) a[i]=c[i];
}


int main()
{
    cin>>T;
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++) scanf("%d",&a[i]);
        
        sort(a,a+m);
        
        for(int i=0;i<n-1;i++)
        {
            for(int j=0;j<m;j++) scanf("%d",&b[j]);
            merge();
        }
            
        for(int i=0;i<m;i++) printf("%d ",a[i]);
        puts("");
    }
    
    return 0;
}