BZOJ 1116 [POI2008]CLO-Toll 并查集

#include <bits/stdc++.h> 
#define N 100005 
#define M 200005 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;    
struct G 
{
    int edges; 
    int hd[N],to[M<<1],nex[M<<1];       
    void add(int u,int v) 
    {
        nex[++edges]=hd[u],hd[u]=edges,to[edges]=v;  
    }
}gr,tr;   
struct Union_Find_set
{ 
    int p[N],is[N];     
    void init() 
    {
        for(int i=0;i<N;++i) p[i]=i; 
    } 
    int find(int x) 
    {
        return p[x]==x?x:p[x]=find(p[x]); 
    }
}ufs;
int n,m,flag;    
int vis[N];     
void dfs(int u,int ff) 
{
    vis[u]=1;    
    for(int i=gr.hd[u];i;i=gr.nex[i]) 
    {
        int v=gr.to[i];     
        if(v==ff) continue;  
        if(!vis[v]) tr.add(u,v),tr.add(v,u),dfs(v, u); 
        else if(vis[v] && flag==0) 
        { 
            tr.add(u, v);    
            tr.add(v,u);    
            flag=1; 
        }       
    }
}
int main()  
{  
    int i,j,k; 
    scanf("%d%d",&n,&m);  
    ufs.init();  
    for(i=1;i<=m;++i) 
    {
        int a,b;  
        scanf("%d%d",&a,&b); 
        gr.add(a,b);  
        gr.add(b,a);     
        a=ufs.find(a); 
        b=ufs.find(b);  
        if(a!=b) ufs.p[a]=b, ufs.is[b]|=ufs.is[a];     
        else ufs.is[a]=1;
    }     
    for(i=1;i<=n;++i) 
    {
        if(ufs.p[i]==i && !ufs.is[i]) 
        {
            printf("NIE\n"); 
            return 0; 
        }
    }
    printf("TAK\n");  
    return 0; 
}