C - 食物链 并查集

这个题目是看别人的博客理解之后再敲的,感觉他的思路很有意思,我也就在自己博客记录一下学习的过程。
**

先看题目吧!

**

动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。

现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。

有人用两种说法对这N个动物所构成的食物链关系进行描述:

第一种说法是"1 X Y",表示X和Y是同类。

第二种说法是"2 X Y",表示X吃Y。

此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。

1) 当前的话与前面的某些真的话冲突,就是假话;

2) 当前的话中X或Y比N大,就是假话;

3) 当前的话表示X吃X,就是假话。

你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。

Input

第一行是两个整数N和K,以一个空格分隔。

以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。

若D=1,则表示X和Y是同类。

若D=2,则表示X吃Y。

Output

只有一个整数,表示假话的数目。

Sample Input
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5

Sample Output
3

题解:

因为一共就只有N个动物,所以我们把数组开大,在N~2N 存某动物吃的动物,在2N~3N存被某动物吃的动物

代码也有注释 比较好理解 直接放代码把!!

#include 
#include
#include 
#include

using namespace std;
int pre[150015];
int N,K,D;
int find(int a)
{
 return pre[a]==a?a:pre[a]=find(pre[a]);
 } 
 void combine(int a,int b)
 {
  int x=find(a);
  int y=find(b);
  if(x!=y)
  {
   pre[x]=y;
  }
 }
 bool pan(int a,int b)//判断是否为同一父亲; 
 {
  int x=find(a);
  int y=find(b);
  if(x==y)
  return true;
  else 
  return false;
 }
int main()
{
 scanf("%d%d",&N,&K);
 for(int i=1;i<=3*N;i++)
 pre[i]=i;
 int ans=0,b,x,y;
 while(K--)
 {
  scanf("%d%d%d",&b,&x,&y);
  if(x>N||y>N)
  {ans++;
  continue;
  } //x+N 表示x被吃的物种,x+2*N代表吃x的物种 
  if(b==1)
  {
   if(pan(x+N,y)||pan(x+2*N,y))//第一个pan是判断y是不是吃x,第二个是判断x是不是吃y
   {
    ans++;
   }
   else//如果不是的话,那就x,y更新为同种物种 
   {
    combine(x,y);
    combine(x+N,y+N);
    combine(x+2*N,y+2*N);
   }
  }
  if(b==2)
  {
   if(pan(x,y)||pan(x+2*N,y))//判断x,y是不是同类 或者 y是不是吃x 
   {
    ans++;
   }
   else//如果不是的话,更新x吃y  3个表达式
   {
    combine(x,y+2*N);
    combine(x+N,y);
    combine(x+2*N,y+N); 
   }
  }
 }
 printf("%d\n",ans);
  return 0;
}

还有好多东西都落下了,要加油呀!!冲鸭!

你可能感兴趣的:(C - 食物链 并查集)