CF Gym 100187J Deck Shuffling （dfs判连通）

#include<cstdio>
#include<vector>
using namespace std;
const int maxn = 200000+4;
int a[maxn];
vector<int> son[maxn];
int x;
bool vis[maxn];
bool dfs(int u)
{
if(a[u] == x) return true;
for(int i = 0; i < son[u].size(); i++){
int v = son[u][i];
if(!vis[v]){
vis[v] = 1;
if(dfs(v)) return true;
}
}
return false;
}

int main()
{
//freopen("in.txt","r",stdin);
int n;
scanf("%d",&n);
for(int i = 0; i < n; i++){
scanf("%d",a+i);
}
int k;
scanf("%d",&k);
for(int i = 0; i < k; i++){
for(int j = 0; j < n; j++){
int t;
scanf("%d",&t);
if(t-1!=j)
son[j].push_back(t-1);
}
}
scanf("%d",&x);
printf("%s",dfs(0)?"YES":"NO");
return 0;
}