HDU4857逃生 拓扑排序+优先级队列

逃生

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Problem Description

糟糕的事情发生啦，现在大家都忙着逃命。但是逃命的通道很窄，大家只能排成一行。

现在有n个人，从1标号到n。同时有一些奇怪的约束条件，每个都形如：a必须在b之前。
同时，社会是不平等的，这些人有的穷有的富。1号最富，2号第二富，以此类推。有钱人就贿赂负责人，所以他们有一些好处。

负责人现在可以安排大家排队的顺序，由于收了好处，所以他要让1号尽量靠前，如果此时还有多种情况，就再让2号尽量靠前，如果还有多种情况，就让3号尽量靠前，以此类推。

那么你就要安排大家的顺序。我们保证一定有解。

 

Input

第一行一个整数T(1 <= T <= 5),表示测试数据的个数。
然后对于每个测试数据，第一行有两个整数n(1 <= n <= 30000)和m(1 <= m <= 100000)，分别表示人数和约束的个数。

然后m行，每行两个整数a和b，表示有一个约束a号必须在b号之前。a和b必然不同。

 

Output

对每个测试数据，输出一行排队的顺序，用空格隔开。

 

Sample Input

1 5 10 3 5 1 4 2 5 1 2 3 4 1 4 2 3 1 5 3 5 1 2

 

Sample Output

1 2 3 4 5

解析：本题使用拓扑排序+优先级队列。

不过此题要求小的数尽量往前放，如果直接用按最小值优先级队列进行拓扑排序是不正确的，比如序列：（4，2），（3，2），（4，1），如果正常使用拓扑排序的话，得到的序列是3->4->1->2，但是正确顺序是4->1->3->2，因为尽可能将小的数放前面，即将1尽可能往前放。

所以得换种思路：反向建图，然后按照最大值优先级队列排序。

代码：

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <vector>
#include <queue>
using namespace std;
#define N 30005
vector<int> V[N];
priority_queue<int> Q;
int cnt[N];
int ans[N];
int main()
{
	int t, n, m;
	int a, b;
	cin>>t;
	while(t--){
		cin>>n>>m;
		for(int i = 1; i <= n; i++)
			V[i].clear();
		while(!Q.empty()) Q.pop();
		memset(cnt, 0, sizeof(cnt));
		for(int i = 0; i < m; i++){
			scanf("%d%d", &a, &b);
			V[b].push_back(a);
			cnt[a]++;
		}
		for(int i = 1; i <= n; i++)
			if(cnt[i] == 0)
				Q.push(i);
		int num = 0;
		while(!Q.empty()){
			int x = Q.top();
			Q.pop();
			ans[num++] = x;
			for(int i = 0; i < V[x].size(); i++){
				int y = V[x].at(i);
				cnt[y]--;
				if(cnt[y]==0)
					Q.push(y);
			}
		}
		for(int i = n-1; i >0; i--)
			printf("%d ", ans[i]);
		printf("%d\n", ans[0]);
	}
	return 0;
}

被这题坑了好多次，还是自己太弱。