PAT | A1146 Topological Order

#include <iostream>
#include <vector>
#include <unordered_set>

using namespace std;

bool G[1010][1010];
int topo[1010];
unordered_set<int> vis;
vector<int> res;

int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i = 0;i < 1010;i++)
		for(int j = 0;j < 1010;j++)
			G[i][j] = false;
	for(int i = 0;i < m;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		G[u][v] = true;
	}
	int k;
	scanf("%d",&k);
	for(int i = 0;i < k;i++){
		vis.clear();
		bool t = true;
		for(int j = 0;j < n;j++){
			scanf("%d",&topo[j]);
			// 看有没有未访问过但指向topo[j]的点
			vis.insert(topo[j]);
			for(int c = 1;c <= n && t == true;c++){
				if(G[c][topo[j]] == true && vis.find(c) == vis.end()){
					t = false;
					break;
				}
			}
		}
		if(t == false)
			res.push_back(i);
	}
	int size = res.size();
	for(int i = 0;i < size;i++){
		printf("%d",res[i]);
		if(i != size - 1)
			printf(" ");
	}
	system("pause");
	return 0;
}