hihoCoder - 1174 - 拓扑排序·一

#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

为了整理一下模板，分别用dfs和直观做法做了一次，每次都是没clear vector错了一发，真傻。。

直观做法和书上介绍的一样，就是用队列模拟。。

AC代码（dfs）：

#include <map>
#include <set>
#include <cmath>
#include <deque>
#include <queue>
#include <stack>
#include <cstdio>
#include <cctype>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 0x7fffffff
using namespace std;

int n, m;

const int maxn = 100005;
vector<int> G[maxn];

int vis[maxn];

bool dfs(int u) {
	vis[u] = -1;
	int d = G[u].size();
	for(int i = 0; i < d; i ++) {
		int v = G[u][i];
		if(vis[v] == -1) {
			return false;
		}
		if(!vis[v] && !dfs(v)) return false;
	}
	vis[u] = 1;
	return true;
}

bool toposort() {
	for(int i = 1; i <= n; i ++) {
		if(!vis[i]) {
			if(!dfs(i)) return false;
		}
	}
	return true;
}

int main() {
	int T;
	scanf("%d", &T);
	while(T --) {
		scanf("%d %d", &n, &m);
		
		for(int i = 1; i <= n; i ++) {
			G[i].clear();
		}
		memset(vis, 0, sizeof(vis));
		
		for(int i = 0; i < m; i ++) {
			int u, v;
			scanf("%d %d", &u, &v);
			G[u].push_back(v);
		}
		
		if(!toposort()) {
			printf("Wrong\n");
		}
		else {
			printf("Correct\n");
		}
	}
	return 0;
}

AC代码（直观做法）：

#include <map>
#include <set>
#include <cmath>
#include <deque>
#include <queue>
#include <stack>
#include <cstdio>
#include <cctype>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 0x7fffffff
using namespace std;

const int maxn = 100005;
int indeg[maxn];
int outdeg[maxn];

vector<int> G[maxn];
int n, m;

bool bfs() {
	int yu = n;
	queue<int> que;
	for(int i = 1; i <= n; i ++) {
		if(indeg[i] == 0) {
			que.push(i);
			yu --;
		}
	}
	
	while(!que.empty()) {
		int u = que.front();
		que.pop();
		int d = G[u].size();
		for(int i = 0; i < d; i ++) {
			int v = G[u][i];
			indeg[v] --;
			outdeg[u] --;
			if(indeg[v] == 0) {
				que.push(v);
				yu --;
			}
		}
	}
	
	if(yu > 0) return false;
	return true;
}

int main() {
	int T;
	scanf("%d", &T);
	while(T --) {
		memset(indeg, 0, sizeof(indeg));
		memset(outdeg, 0, sizeof(outdeg));
		scanf("%d %d", &n, &m);
		for(int i = 1; i <= n; i ++) {
			G[i].clear();
		}
		for(int i = 0; i < m; i ++) {
			int u, v;
			scanf("%d %d", &u, &v);
			G[u].push_back(v);
			outdeg[u] ++;
			indeg[v] ++;
		}
		
		if(bfs()) {
			printf("Correct\n");
		}
		else {
			printf("Wrong\n");
		}
	}
	return 0;
}