ACM: hihicoder #1174 : 拓扑排序·一 STL- queue

#1174 : 拓扑排序·一

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

由于今天上课的老师讲的特别无聊，小Hi和小Ho偷偷地聊了起来。

小Ho：小Hi，你这学期有选什么课么？

小Hi：挺多的，比如XXX1,XXX2还有XXX3。本来想选YYY2的，但是好像没有先选过YYY1，不能选YYY2。

小Ho：先修课程真是个麻烦的东西呢。

小Hi：没错呢。好多课程都有先修课程，每次选课之前都得先查查有没有先修。教务公布的先修课程记录都是好多年前的，不但有重复的信息，好像很多都不正确了。

小Ho：课程太多了，教务也没法整理吧。他们也没法一个一个确认有没有写错。

小Hi：这不正是轮到小Ho你出马的时候了么！

小Ho：哎？？

我们都知道大学的课程是可以自己选择的，每一个学期可以自由选择打算学习的课程。唯一限制我们选课是一些课程之间的顺序关系：有的难度很大的课程可能会有一些前置课程的要求。比如课程A是课程B的前置课程，则要求先学习完A课程，才可以选择B课程。大学的教务收集了所有课程的顺序关系，但由于系统故障，可能有一些信息出现了错误。现在小Ho把信息都告诉你，请你帮小Ho判断一下这些信息是否有误。错误的信息主要是指出现了"课程A是课程B的前置课程，同时课程B也是课程A的前置课程"这样的情况。当然"课程A是课程B的前置课程，课程B是课程C的前置课程，课程C是课程A的前置课程"这类也是错误的。

提示：拓扑排序

输入

第1行：1个整数T，表示数据的组数T(1 <= T <= 5)
接下来T组数据按照以下格式：
第1行：2个整数，N,M。N表示课程总数量，课程编号为1..N。M表示顺序关系的数量。1 <= N <= 100,000. 1 <= M <= 500,000
第2..M+1行：每行2个整数，A,B。表示课程A是课程B的前置课程。

输出

第1..T行：每行1个字符串，若该组信息无误，输出"Correct"，若该组信息有误，输出"Wrong"。

样例输入

2
2 2
1 2
2 1
3 2
1 2
1 3

样例输出

Wrong
Correct

/*/
拓扑排序的第一题，为了理解拓扑排序一开始用邻接矩阵写了一个，发现跟题目说的一样会爆内存，后面发现，STL的队列更加简单易懂，果断写了一个A题。

AC代码：
/*/

#include"algorithm"
#include"iostream"
#include"cstring"
#include"string"
#include"cstdio"
#include"vector"
#include"cmath"
#include"queue"
using namespace std;
#define memset(x,y) memset(x,y,sizeof(x))
#define MX 100000 + 10

/***********************************************************
				超时，超内存-写法-邻接矩阵
***********************************************************////
int cnt;
int map[MX][MX]; //邻接矩阵
int indegree[MX]; //入度

void toposort(int map[MX][MX],int indegree[MX],int n,int m){
	for(int i=1;i<=m;i++){//进行m次历遍
		for(int j=1;j<=n;j++){	//j 枚举每一个点
			if(!indegree[j]){	//如果这个点入度为0，标记已经查过了。
				indegree[j]--;
				for(int k=1;k<=n;k++){	//检查这个点的其他边，并且消掉这些边。
					if(map[j][k]==1){
						indegree[k]--; //入度--
						cnt--; //清除该条有向边
					}
				}
			}
		}
	}
}

int main() {
	int T;
	int n,m,st,ed;
	scanf("%d",&T);
	while(T--) {
		scanf("%d%d",&n,&m);
		cnt=0;
		memset(map,0);
		memset(indegree,0);
		for(int i=0; i<m; i++) {
			scanf("%d%d",&st,&ed);
			map[st][ed]=1;	//该有向边存在
			indegree[ed]++; //入度++
			cnt++;//有向边的数量
		}
		toposort(map,indegree,n,m);
		cout<<cnt<<endl;
		if(cnt==0) puts("Correct"); //如果有向边数变为0 则说明是正确的
		else puts("Wrong");
	}
	return 0;
}
//*********************************************************/

/***********************************************************
				        queue
***********************************************************/

int T,n,m,head[MX],cnt,p[MX];
queue<int>q;

struct node {
	int e,next;
} edge[MX];

void add(int st,int ed) {
	edge[cnt].e=ed;
	edge[cnt].next=head[st];
	head[st]=cnt++;
}

int main() {
	int i,st,ed,ans;
	scanf("%d",&T);
	while(T--) {
		ans=0;
		cnt=1;
		scanf("%d%d",&n,&m);
		memset(p,0);
		memset(head,-1);
		for(i=0; i<m; i++) {
			scanf("%d%d",&st,&ed);
			add(st,ed);
			p[ed]++;
		}
		for(i=1; i<=n; i++)   //把入度为0的压入队列
			if(!p[i]) {
				q.push(i);
				ans++;
			}
		while(!q.empty()) {
			int k=q.front();
			q.pop();
			for(i=head[k]; i!=-1; i=edge[i].next) {
				int v=edge[i].e;
				p[v]--;
				if(!p[v]) {   //把入度为0的压入队列
					q.push(v);
					ans++;		//标记入度为0的数量
				}
			}
		}
		if(ans==n)puts("Correct"); //如果入度为0的节点等于总点数 则说明是正确的
		else puts("Wrong");
	}
	return 0;
}
//*********************************************************/