WEEK8 作业 B - 猫猫向前冲

B - 猫猫向前冲

题目描述

众所周知， TT 是一位重度爱猫人士，他有一只神奇的魔法猫。
有一天，TT 在 B 站上观看猫猫的比赛。一共有 N 只猫猫，编号依次为1，2，3，…，N进行比赛。比赛结束后，Up 主会为所有的猫猫从前到后依次排名并发放爱吃的小鱼干。不幸的是，此时 TT 的电子设备遭到了宇宙射线的降智打击，一下子都连不上网了，自然也看不到最后的颁奖典礼。
不幸中的万幸，TT 的魔法猫将每场比赛的结果都记录了下来，现在他想编程序确定字典序最小的名次序列，请你帮帮他。

Input

输入有若干组，每组中的第一行为二个数N（1<=N<=500），M；其中N表示猫猫的个数，M表示接着有M行的输入数据。接下来的M行数据中，每行也有两个整数P1，P2表示即编号为 P1 的猫猫赢了编号为 P2 的猫猫。

Output

给出一个符合要求的排名。输出时猫猫的编号之间有空格，最后一名后面没有空格!

其他说明：符合条件的排名可能不是唯一的，此时要求输出时编号小的队伍在前；输入数据保证是正确的，即输入数据确保一定能有一个符合要求的排名。

Sample Input

4 3
1 2
2 3
4 3

Sample Output

1 2 4 3

题解

这道题用拓扑排序的方法来解。对于每只猫，看作图中的点，A赢了B，即为图中从A指向B的一条有向边。入度为0的点，就是没有输过的猫。
因为要找字典序最小的排序，所以我们使用的是优先级队列，队列中的元素优先取出序号最小的。
首先遍历所有元素的入度，将入度为0的元素加入队列。循环，若队列不为空，从队列中取出首元素，加入到结果数组v，将该元素指向的所有元素的入度-1，并检查是否为0，为0则加入队列，然后进入下一次循环。

代码

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std; 
int n;
int in_deg[510];
int head[510];
int tot=0;
struct edge{
	int to,nxt;
}e[510];
void addEdge(int x,int y){
	e[tot].nxt=head[x];
	e[tot].to=y;
	head[x]=tot;
	tot++;
}
int main(int argc, char** argv) {
	int m;
	while(scanf("%d%d",&n,&m)!=EOF){
		tot=0;
		memset(in_deg,0,sizeof(in_deg));
		memset(head,-1,sizeof(head));
		while(m--){
			int p1,p2;
			scanf("%d%d",&p1,&p2);
			in_deg[p2]++;
			addEdge(p1,p2);
		}
		priority_queue<int,vector<int>,greater<int> >pq;
		vector<int> v;
		for(int i=1;i<=n;i++){
			if(in_deg[i]==0){pq.push(i);}
		}
		while(!pq.empty()){
			int x=pq.top();
			v.push_back(x);
			pq.pop();
			int next=head[x];
			while(next!=-1){
				if(--in_deg[e[next].to]==0){
					pq.push(e[next].to);
				}
				next=e[next].nxt;
			}
		}
		for(int i=0;i<n;i++){
			if(i==n-1){
				printf("%d\n",v[i]);
			}else{
				printf("%d ",v[i]);
			}
		}
	}
	return 0;
}