首页 > 代码库 > POJ 1703 Find them, catch them (并查集)

POJ 1703 Find them, catch them (并查集)

  题目: Find them,Catch them

  刚开始以为是最基本的并查集,无限超时。

  这个特殊之处,就是可能有多个集合。

  比如输入D 1 2  D 3 4 D 5 6...这就至少有3个集合了。并且任意2个集合之间成员的敌我关系不明。

  这里每个集合里面的成员关系要记录,他们在一个集合里,只是因为他们关系明确,敌人或者朋友。

  千万不要简单的认为成朋友在一个集合,敌人在另外一个集合,分成2个集合。因为题目说只有2个匪帮,很容易进入这个误区。

  我们只要记录一个成员和自己父亲的敌我关系和建树就可以了。

  

代码:

#include<iostream>#include<cstdio>using namespace std;const int MM = 100001;int father[MM];int relation[MM];int find(int x){    if(x == father[x]) return x;    int r = find(father[x]);    relation[x] ^= relation[father[x]];    return father[x] = r;}void join(int x, int y){    int fx = find(x);    int fy = find(y);        father[fx] = fy;    if(relation[y] == 1)        relation[fx] = relation[x];     else        relation[fx] = 1 - relation[x];}void solve(){    int T,N,M,i,a,b,ta,tb;    char str;    scanf("%d", &T);    while(T--)    {        scanf("%d %d",&N, &M);                for(i = 1; i <= N; ++i)        {            father[i] = i;            relation[i] = 0;        }                for(i = 0; i < M; ++i)        {            getchar();            scanf("%c %d %d", &str, &a, &b);            if(str == D)                join(a, b);            else            {                ta = find(a);                tb = find(b);                if(ta == tb)                {                    if(relation[a] == relation[b])                        printf("In the same gang.\n");                    else                        printf("In different gangs.\n");                }                   else                    printf("Not sure yet.\n");            }        }    }}int main(){    solve();    return 0;}

 

POJ 1703 Find them, catch them (并查集)