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