码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 1703 Find them, Catch them

时间:2015-10-18 15:18:11      阅读:318      评论:0      收藏:0      [点我收藏+]

标签:

DisjointSet保存的是等价关系,对于某个人X,设置两个变量Xa,Xb。Xa表示X属于a帮派,Xb类似。

如果X和Y不是同一个帮派,那么Xa -> Yb,Yb -> Xa...

(Xa,Yb)对称性,传递性,自反都满足。所以可以用合并到一个集合里面。

判断就比较简单了。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
//#include<bits/stdc++.h>
using namespace std;


const int maxn = 2e5+2;
int pa[maxn];
int hei[maxn];

int fdst(int x){ return x==pa[x]?x:pa[x]=fdst(pa[x]); }
inline void mrge(int a,int b)
{
    int s1 = fdst(a), s2 = fdst(b);
    if(s1 != s2){
        if(hei[s1] < hei[s2])
            pa[s1] = s2;
        else {
            pa[s2] = s1;
            if(hei[s1] == hei[s2]) hei[s1]++;
        }
    }
}
inline bool same(int a,int b){ return fdst(a) == fdst(b); }

void initDisjointSet(int n)
{
    for(int i = 1; i <= n; i++){
        pa[i] = i;
        hei[i] = 0;
    }
}

//#define LOCAL
int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif
    int T; cin>>T;
    while(T--){
        int n, m; scanf("%d%d",&n,&m);
        initDisjointSet(n*2);
        while(m--){
            char ch[2];
            int a,b;
            scanf("%s%d%d",ch,&a,&b);
            if(*ch == D){
                mrge(a,b+n);
                mrge(a+n,b);
            }else {
                if(same(a,b+n)){
                    puts("In different gangs.");
                }else if(same(a,b)){
                    puts("In the same gang.");
                }else puts("Not sure yet.");
            }
        }
    }
    return 0;
}

 

POJ 1703 Find them, Catch them

标签:

原文地址:http://www.cnblogs.com/jerryRey/p/4889468.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!