codeforces Gym 100187J J. Deck Shuffling dfs

题目链接:

http://codeforces.com/gym/100187/problem/J

题意:

给你一堆牌，和一些洗牌机，可以改变牌的顺序，问你能不能通过洗牌机把数字为x的牌洗到第一个位置。
样例一： 最初的牌 4 3 2 1 通过第一个洗牌机把第四个位置的x(=1)洗到第三个位置 然后 第二个洗牌机把当前在第三个位置x洗到第一个位置

题解:

建边，把洗牌机每个位置–>下一个位置（也就是这个位置的值）
跑一发dfs就好了，其实就问你第一个位置和x所在的位置是否联通

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 2e5+10;

int a[maxn],vis[maxn];
map<pair<int,int>, int> mp;
vector<int> e[maxn];
bool f;

void dfs(int x){
    if(x == 1){
        f = true;
        return ;
    }
    if(f) return ;
    if(vis[x]) return ;

    vis[x] = 1;

    for(int i=0; i<(int)e[x].size(); i++){
        if(vis[e[x][i]])
            continue;
        dfs(e[x][i]);
    }
}

int main(){
    int n = read();
    for(int i=1; i<=n; i++){
        int x = read();
        a[x] = i;
    }

    int k = read();
    for(int i=0; i<k; i++){
        for(int j=1; j<=n; j++){
            int x = read();
            // if(mp[MP(j,x)]!=1){
            //  mp[MP(j,x)] = 1;
            //  e[j].PB(x);
            // }
            e[j].PB(x);  
        }
    }

    int x = read();
    dfs(a[x]);

    if(f) puts("YES");
    else puts("NO");

    return 0;
}