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;
}