程序设计Week8——B-猫猫向前冲

题目描述

有n只猫两两比赛，先记录了每场比赛的胜负结果，要求给出猫猫的排名，胜者在前，败者在后，保证最小字典序。

Input

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

Output

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

解题思路

拓扑排序题，而且要输出字典序最小的拓扑序，直接用Kahn算法求解，将其中的队列换为最小堆即可。
Kahn算法：

• 所有入度为0的点加入队列；
• 每次从队列中取出一个点u，放入ans数组，然后遍历与这个点u邻接的点v，判断点v在删掉与u相连的这条边后，入度是否为0，若为0就将点v也加入队列；
• 如此重复直到整个队列为空，检查此时是不是还有边（或者是ans中的点有无重复），若有则说明有环，那么无法进行拓扑排序，否则ans数组就是该图的拓扑序。

实现代码

#include<iostream>
#include<queue>
#include<vector>
#include<cstring>
#include<functional>
using namespace std;

int degree[520];
vector<int> G[520],ans;
int n,m;

void init()
{
	memset(degree,0,n);
	ans.clear();
	for(int i=1;i<=n;++i)
		G[i].clear();
}

void solve()
{
	priority_queue<int,vector<int>,greater<int> > q;
	//int t=0;
	for(int i=1;i<=n;++i)
	{
		if(degree[i]==0)
			q.push(i);
	}
	while(!q.empty())
	{
		int u=q.top();
		q.pop();
		ans.push_back(u);
		for(int i=0;i<G[u].size();++i)
		{
			int v=G[u][i];
			degree[v]--;
			if(degree[v]==0)
				q.push(v);
		}
	}
	for(int i=0;i<ans.size();++i)
	{
		cout<<ans[i];
		if(i!=ans.size()-1)
			cout<<" ";
	}
}

int main()
{
	while(cin>>n>>m)
	{
		init();
		for(int i=1;i<=m;++i)
		{
			int a,b;
			cin>>a>>b;
			degree[b]++;
			G[a].push_back(b);
		}
		solve();
		cout<<endl;
	}
	return 0;
} 

总结

本题保证了数据一定合法，因此就无需判定环路了，求字典序最小的拓扑序直接利用小根堆即可。
但这里提交测试的时候，本地运行没毛病，但是评测机却CE了，原因是小根堆定义错误（？？？），百度之后加上了functional这个头文件才过了，但是这个问题我还是第一次碰见，之前小根堆也没被这样欺负过！当然原因我也不太清楚，下图是CE图：