ZOJ 3811 dfs&并查集

点击打开链接

题意：给n个城镇，有的城镇会发光，给出发光的城镇，然后下面有一个人走的发光的城镇的顺序，每个城镇只会发一次光，问这个人按这个顺序是否合法并且所有的城镇都被访问到

思路：可以用dfs去搜索，第一个点进去，不能走的点标记一下，然后第二个发光的城镇，判断目前标记的点中有没有，有的话说明现在可以走第二个点，以此类推，最后判断城镇是否全被访问过，一会试试并查集的方法

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int vis1[maxn],vis2[maxn],id[maxn],vis[maxn],ansvis[maxn];
vector<int>G[maxn];
void dfs(int u){
    if(vis[u]) return ;
    vis[u]=1;
    for(unsigned int i=0;i<G[u].size();i++){
        int t=G[u][i];
        if(vis[t]) continue;
        if(vis2[t]) ansvis[t]=1;
        else dfs(t);
    }
}
int main(){
    int T,n,m,k,u,v,l;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<maxn;i++) G[i].clear();
        memset(vis1,0,sizeof(vis1));
        memset(vis2,0,sizeof(vis2));
        memset(vis,0,sizeof(vis));
        memset(ansvis,0,sizeof(ansvis));
        for(int i=0;i<k;i++){
            scanf("%d",&u);vis1[u]=1;
        }
        for(int i=0;i<m;i++) {
            scanf("%d%d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        scanf("%d",&l);
        for(int i=0;i<l;i++){
            scanf("%d",&id[i]);vis2[id[i]]=1;
        }
        int sum=0;
        for(int i=0;i<k;i++) if(vis1[id[i]]&&vis2[id[i]]) sum++;
        if(sum!=k||l!=k){
            printf("No\n");
            continue;
        }
        vis2[id[0]]=0;dfs(id[0]);
        int flag=0;
        for(int i=1;i<l;i++){
            if(ansvis[id[i]]){
                vis2[id[i]]=0;
                dfs(id[i]);
            }else{
                flag=1;break;
            }
        }
        for(int i=1;i<=n;i++) if(vis[i]==0) flag=1;
        if(flag) printf("No\n");
        else printf("Yes\n");
    }
    return 0;
}

是想到并查集的方法了，太麻烦不想写了，而且我的想法还是要基于DFS的，所以不如就只有dfs了，太弱~~~