UVA 11997之多路归并问题求解

看了下同学的算法竞赛入门经典这本书，正好是UVA的题，于是就按照那个上面的顺序开始刷，看到一道题是LA上的，大概说的是有两个信号源分别按照各自周期发出信号，每个都有相应的周期，然后还有各自ID，然后让你根据发出时间和ID来输出信号源发出信号的序列，感觉类似于CPU的调度问题，就是维护一个优先队列，每次push，pop就行了。

然后书中提到这题的思路可以延伸到多路归并问题的求解中，随后的一题便是多路归并问题，假设你有多个序列，并且假设每个序列都是排好序的，那么就可以将每个序列的序列头放在一个优先队列中，然后维护这个优先队列即可，每次取出队列头并push进一个数，因为序列是排序好的，这样就能保证取出的数据成为一个新的排好序的列表，代码如下：

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

using namespace std;

struct item{
int sum,b;
item(int sum, int b):sum(sum), b(b){}
bool operator <(const item& it) const{
return sum > it.sum;
}
};

void merge(int *A, int *B, int *C, int n){
priority_queue<item> que;
for(int i=0; i<n; i++)
que.push(item(A[i]+B[0], 0));
for(int i=0; i<n; i++){
item it = que.top();
que.pop();
C[i] = it.sum;
int b = it.b;
if(b+1<n)
que.push(item(it.sum-B[b]+B[b+1], b+1));
}

}

int A[751][751];

int main(){
int n;
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);
}
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");
}
return 0;
}