poj1703

本题主要算法并查集，一般解决集合合并，是否在同一集合，一个集合中的元素个数等问题。复杂一点的就是带偏移量的，本题和1182很像，但是这个是只有两种情况。下面有个优化的算法。但是不是太多。

//594MS
/*#include<iostream>
#define N 100010
using namespace std;
int n,f[N],rank[N];

int cal(int a)
{
 int t=0;
 while(f[a]!=-1)
 {
  t+=rank[a];
  a=f[a];
 }
 return t;
}

int Find(int t)
{
 while(f[t]!=-1)
  t=f[t];
 return t;
}

void Union(int a,int b)
{
 int x,y;
 x=Find(a);
 y=Find(b);
 if(cal(a)<cal(b))
 {
  f[x]=y;
  rank[x]=(cal(a)-cal(b)+1)%2;
 }
 else
 {
  f[y]=x;
  rank[y]=(cal(a)-cal(b)+1)%2;
 }
}

int main()
{
 int i,j,k,m,x,y;
 char ch;
 scanf("%d",&k);
 while(k--)
 {
  scanf("%d%d",&n,&m);
  memset(f,-1,sizeof(f));
  memset(rank,0,sizeof(rank));
  while(m--)
  {
   getchar();
   scanf("%c %d %d",&ch,&x,&y);
   
   if(ch=='A')
   {
    if(Find(x)!=Find(y))
    {
     printf("Not sure yet./n");
     
    }
    else
    {
     if((cal(x)-cal(y))%2==0)
      printf("In the same gang./n");
     else
      printf("In different gangs./n");
    }
   }
   else if(ch=='D')
   {
    if(Find(x)!=Find(y))
     Union(x,y);
   }
  }
 }
 return 0;
}*/

 

我优化了一下。

//391MS
#include<iostream>
#define N 100010
using namespace std;
int n,f[N],rank[N];

void Init()
{
 for(int i=1;i<=n;i++)
 {
  f[i]=i;
  rank[i]=0;
 }
}

int Find(int x)
{
 if(x==f[x])
  return x;
 int tmp=Find(f[x]);
 
 rank[x]=(rank[x]+rank[f[x]])%2;//由于最终的根节点rank为0，所以不会改变偏移量，反而带偏移量下面的子节点就会相应改变
 f[x]=tmp;
 return f[x];
}

void Union(int a,int b)
{
 int x,y;
 x=Find(a);
 y=Find(b);
 f[x]=y;
 rank[x]=(rank[a]+rank[b]+1)%2;//修改根存偏移量，在find的 时候依次更新子节点。
}

int main()
{
 int k,m,x,y;
 char ch;
 scanf("%d",&k);
 while(k--)
 {
  
  scanf("%d%d",&n,&m);
  Init();
  while(m--)
  {
   getchar();
   scanf("%c %d %d",&ch,&x,&y);
  /* //时间没什么改变，如果改成下面的代码。
   if(ch=='A')
   {
    if(Find(x)!=Find(y))
    {
     printf("Not sure yet./n");
     
    }
    else
    {
     if(rank[x]==rank[y])
      printf("In the same gang./n");
     else
      printf("In different gangs./n");
    }
   }
   else if(ch=='D')
   {
    if(Find(x)!=Find(y))
     Union(x,y);
   }*/
   if(Find(x)!=Find(y))
   {
    if(ch=='A')
    {
     printf("Not sure yet./n");
     continue;
    }
    else
    {
     Union(x,y);
    } 
   }
   else
   {
    if(ch=='A')
    {
     if(rank[x]==rank[y])
      printf("In the same gang./n");
     else
      printf("In different gangs./n");
    }
   }

  }
 }
 return 0;
}