序列合并求前K小项 POJ2442

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

using namespace std;

priority_queue<int>pq;
int an[3100];
int bn[3100];

int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int m,n;
        scanf("%d%d",&m,&n);
        for(int i=0;i<n;i++)
            scanf("%d",&an[i]);
        for(int i=1;i<m;i++)
        {
            for(int t=0;t<n;t++)
            {
                scanf("%d",&bn[t]);
            }
            for(int t=0;t<n;t++)
            {
                pq.push(an[0]+bn[t]);
            }
            for(int t=1;t<n;t++)
            {
                for(int j=0;j<n;j++)
                {
                    if(an[t]+bn[j]<pq.top())
                    {
                        pq.pop();
                        pq.push(an[t]+bn[j]);
                    }
                }
            }
            for(int t=0;t<n;t++)
            {
                an[t]=pq.top();
                pq.pop();
            }
        }
        cout<<an[n-1];
        for(int i=n-2;i>=0;i--)
            cout<<" "<<an[i];
        cout<<endl;
    }
    return 0;
}

解题步骤：

1.将第一序列读入data1向量中，并按升序排序。

2.将数据读入data2向量中，并按升序排序。

    将data2[0] + data1[i] ( 0<=i<=n-1)读入dataq向量中

    用make_heap对dataq建堆。

    然后data2[1] + data1[i] (0<=i<=n-1),如果data2[1] + data1[i]比堆dataq的顶点大,则退出，否则删除

    堆的顶点，插入data2[1] + data1[i]。然后是data2[2],...data2[n - 1]

 3.将dataq的数据拷贝到data1中，并对data1按升序排序

 4.循环2,3步，直到所有数据读入完毕。

 5.打印data1中的数据即为结果。

按此方法，每次只需和当前最小的K项中最大的那项比较就行，而不用比较n^m次