poj1703Find them, Catch them(并查集以及路径压缩)

简介:
/*
   题目大意:有两个不同的黑帮,开始的时候不清楚每个人是属于哪个的!
  执行两个操作
  A a, b回答a, b两个人是否在同一帮派,或者不确定
  D a, b表示a, b两个人不在同一个帮派

  思路:利用并查集将相同帮派的人合并到一起!
           a ,b 不在同一个城市,那么 a, 和mark[b]就在同一个城市,
           b 和 mark[a]就在同一个城市!
*/
#include<iostream>
#include<cstring>
#include<cstdio>
#define M 100005
using namespace std; 
int f[M];
int mark[M];//mark[i]表示 i 与 mark[i]不在同一个黑帮 
int rank[M];//为不同的集合的父亲节点记录其孩子机节点个数,在合并集合的时候,将
int n, m;   //rank 值小的集合连接到 rank 值大的集合上去! 这样在路径压缩的时候省去不少时间 

void init(){
    
   memset(mark, 0, sizeof(int)*(n+1));
   memset(rank, 0, sizeof(int)*(n+1));
   for(int i=1; i<=n; ++i)
      f[i]=i;
}

int getFather(int x){
   return x==f[x] ? x : f[x]=getFather(f[x]); 
}

void Union(int a, int b){
   int fa=getFather(a), fb=getFather(b);
   if(fa!=fb){
      if(rank[fa]>rank[fb]){ 
         f[fb]=fa;
         rank[fa]+=rank[fb]+1;
      }
      else{
         f[fa]=fb;
         rank[fb]+=rank[fa]+1;
      }
   }

int main(){
   int t;
   char cmd[3];
   scanf("%d", &t);
   while(t--){
      scanf("%d%d", &n, &m);
      init();
      while(m--){
         int u, v;
         scanf("%s%d%d", cmd, &u, &v);
         if(cmd[0]=='A'){
              int fu=getFather(u), fv=getFather(v);
             if(fu==fv)
                printf("In the same gang.\n");
             else if(fu==getFather(mark[v]))
                printf("In different gangs.\n"); 
             else
                printf("Not sure yet.\n");
         }
         else{
            if(!mark[u] && !mark[v]){
               mark[u]=v;
               mark[v]=u;
            } 
            else if(!mark[u]){
               mark[u]=v;
               Union(u, mark[v]);
            }
            else if(!mark[v]){
               mark[v]=u;
               Union(v, mark[u]);
            }
            else {
               Union(u, mark[v]);
               Union(v, mark[u]);
            }
         }
      }
   } 
   return 0;
}









本文转自 小眼儿 博客园博客,原文链接:http://www.cnblogs.com/hujunzheng/p/3892571.html,如需转载请自行联系原作者
目录
相关文章
|
8月前
|
网络架构
POJ 3250 Bad Hair Day、POJ 2796 Feel Good(单调栈)
POJ 3250 Bad Hair Day、POJ 2796 Feel Good(单调栈)
HDU-1598,find the most comfortable road(并查集+枚举)
HDU-1598,find the most comfortable road(并查集+枚举)
|
算法 测试技术
【算法笔记题解】PAT A1075 PAT Judge
【算法笔记题解】PAT A1075 PAT Judge
【算法笔记题解】PAT A1075 PAT Judge
|
并行计算
Find a way(两个BFS)
Problem Description Pass a year learning in Hangzhou, yifenfei arrival hometown Ningbo at finally.
1085 0