uva11997 ADT

从m个数组（每个数组中n个元素）中各任取一个元素求和，得到n^m个元素。从小到大输出前k个合并后的元素。

常规做法：将所有元素求和排序，O((n^m)*log(n^m))，超时

巧妙做法：

1.分别将m个数组进行排序，得到m个有序表 O(m*n*logn)

2.每个有序表中各取一个元素相加后排序，求前k小的元素

简化：m=2时，设两个有序表为a[0,1...n-1],b[0,1...n-1],从a和b数组中各取一个元素，求和得到n^2个数，求前k小元素

构造新的有序表,第i个有序表元素为：b[i]+a[j],1<=j<=n-1

则问题转化为归并n个有序表（每个表有n个元素），时间复杂度为O(n*n*logn)，如果只求前k小元素，则时间复杂度降为O(k*logn)

m>2时，可以每两个数组构造出n个有序表，得到前k小元素，再依次和其他数组进行相同处理，总时间复杂度：O(m*k*logn)

时间复杂度下降的原因是只考虑了前k小元素，而未将后面的元素有序化。

注：合并n个有序表，共含m个元素，时间复杂度：O(m*logn)

原理：有序表内部已经有序，因此同时只需要维护有序表大小（n）个数的大小关系，在取出这n个数中的最小值后，再将此有序表中的后一个元素添加到此维护序列中

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

struct Item{
    int s,b;//s=A[a]+B[b]
    Item(int s,int b):s(s),b(b){ }
    bool operator<(const Item& rhs) const{
        return s>rhs.s;
    }
};

void merge(int* A,int* B,int* C,int n){
    priority_queue<Item> q;
    for(int i=0;i<n;i++)
        q.push(Item(A[i]+B[0],0));
    for(int i=0;i<n;i++){
        Item item=q.top();q.pop();//取出A[a]+B[b]
        C[i]=item.s;//合并值的前n个值依次加入
        int b=item.b;
        if(b+1<n) q.push(Item(item.s-B[b]+B[b+1],b+1));//更新s值加入合并表
    }
}

const int maxn=768;
int A[maxn][maxn];

int main(){
    int n;
    //freopen("a.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        //预处理，使数组有序
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++) scanf("%d",&A[i][j]);
            sort(A[i],A[i]+n);
        }
        //两两合并：构造n个有序表，再将其合并
        for(int i=1;i<n;i++)
            merge(A[0],A[i],A[0],n);
        //输出结果
        printf("%d",A[0][0]);
        for(int i=1;i<n;i++) printf(" %d",A[0][i]);
        printf("\n");
    }
}