CF Gym 100187J Deck Shuffling (dfs判连通)

题意:给你一堆牌,和一些洗牌机,可以改变牌的顺序,问你能不能通过洗牌机把数字为x的牌洗到第一个位置。

题解:反向建边,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;
}

 

posted @ 2015-07-19 09:53  陈瑞宇  阅读(336)  评论(0编辑  收藏  举报