这是 20182018 年研究生入学考试中给出的一个问题:
以下哪个选项不是从给定的有向图中获得的拓扑序列?
现在,请你编写一个程序来测试每个选项。
输入格式
第一行包含两个整数 NN 和 MM,分别表示有向图的点和边的数量。
接下来 MM 行,每行给出一条边的起点和终点。
点的编号从 11 到 NN。
再一行包含一个整数 KK,表示询问次数。
接下来 KK 行,每行包含一个所有点的排列。
一行中的数字用空格隔开。
输出格式
在一行中输出所有不是拓扑序列的询问序列的编号。
询问序列编号从 00 开始。
行首和行尾不得有多余空格,保证存在至少一个解。
数据范围
1≤N≤10001≤N≤1000,
1≤M≤100001≤M≤10000,
1≤K≤1001≤K≤100
输入样例:
6 8
1 2
1 3
5 2
5 4
2 3
2 6
3 4
6 4
5
1 5 2 3 6 4
5 1 2 6 3 4
5 1 2 3 6 4
5 2 1 6 3 4
1 2 3 4 5 6
输出样例:
3 4
代码如下:
#include<iostream>
#include<queue>
#include<vector>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int n, m;
vector<int> G[1010];
int in[1010];
int mt[1010];
int a[1010];
vector<int> res;
bool toppsort() {
for(int i = 1; i <= n; i ++) in[i] = mt[i]; //每次都维护一下入度
for(int i = 1; i <= n; i ++) {
if(in[a[i]] != 0) return false; //如果这个点入度不为0,则不是拓扑序列
in[a[i]] --;
for(int j = 0; j < G[a[i]].size(); j ++) {
in[G[a[i]][j]] --; // 每次只对受影响的点的入度进行处理
// 如果用数组,每次对所有边进行判断,会超时。
}
}
return true;
}
int main() {
memset(in, 0, sizeof in);
scanf("%d%d", &n, &m);
int x, y, t;
for(int i = 0; i < m; i ++) { // init
scanf("%d%d", &x, &y);
G[x].push_back(y);
mt[y] ++;
}
scanf("%d", &t);
for(int i = 0; i < t; i ++) {
for(int j = 1; j <= n; j ++) {
scanf("%d", &a[j]);
}
if(!toppsort())
res.push_back(i);
}
for(int i = 0; i < res.size(); i ++) {
printf("%d ", res[i]);
}
return 0;
}