PAT甲级 1146 拓扑顺序

原题链接

这是 2018 年研究生入学考试中给出的一个问题：

以下哪个选项不是从给定的有向图中获得的拓扑序列？

现在，请你编写一个程序来测试每个选项。

输入格式
第一行包含两个整数 N 和 M，分别表示有向图的点和边的数量。

接下来 M 行，每行给出一条边的起点和终点。

点的编号从 1 到 N。

再一行包含一个整数 K，表示询问次数。

接下来 K 行，每行包含一个所有点的排列。

一行中的数字用空格隔开。

输出格式
在一行中输出所有不是拓扑序列的询问序列的编号。

询问序列编号从 0 开始。

行首和行尾不得有多余空格，保证存在至少一个解。

数据范围
1≤N≤1000,
1≤M≤10000,
1≤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 <bits/stdc++.h>
using namespace std;
const int N = 1010, M = 10010;
struct Edge{
    int a, b;
}e[M];
int p[N];
int n, m;
int main(){
    cin >> n >> m;
    for(int i = 0; i < m; i ++ ){
        cin >> e[i].a >> e[i].b;
    }
    int k;
    cin >> k;
    bool is_first = true;
    for(int i = 0; i < k; i ++ ){
        for(int j = 1; j <= n; j ++ ){
            int x;
            cin >> x;
            p[x] = j;
        }
        bool success = true;
        for(int v = 0; v < m; v ++ ){
            if(p[e[v].a] > p[e[v].b]){
                success = false;
                break;
            }
        }
        if(!success){
            if(is_first) is_first = false;
            else cout << " ";
            cout << i;
        }
    }
    return 0;
}