POJ 1182 食物链

。。

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 105599		Accepted: 31966

Description

动物王国中有三类动物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

Source

Noi 01

。。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <stdio.h>
#include <cstring>
#include <string>
#include <cstdlib>
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <map>
#include <list>
#include <iomanip>
 #include <fstream>

using namespace std;
int fa[50009],eat[50009];
//不要忘记初始化数组惹qwq 
void initailize(int n)
{
    for(int i=0;i<n;++i)
    {
        fa[i]=i;//一开始自立门户 
        eat[i]=0;//食性是莫得食性的 
     } 
}
int get(int x)//查询根节点
{
    int temp;
    if(x==fa[x])
        return x;//找到根节点 
        
    temp=fa[x];//没有找到根节点，但是找到了上一级 
    fa[x]=get(temp);//更新自己的根节点，找到了上一级的根节点
    //捕食关系更新，根据自己祖宗的食性知道自己的食性
     eat[x]=(eat[x]+eat[temp])%3;//哇呜，传说中的矢量运算了？ 
     //暂时规定其中的关系数字 0 1 2 分别代表同类/捕食/被捕食 
     return fa[x];
}

int main()
{
    int n,k,cnt=0;
    scanf("%d%d",&n,&k);
    initailize(n);//不知道有没有拼对，反正是初始化的意思 
    for(int i=1;i<=k;++i)
    {
        int d,x,y;
        scanf("%d%d%d",&d,&x,&y);
        //注意特殊情况（条件2、3），小动物的编号都已经超过了n
         if(x>n||y>n||(x==y&&d==2))//我~吃~我~自~己~
             {cnt++;continue;}
        int gx=get(x),gy=get(y);//先找一下自己祖宗 
        //如果他们祖宗不一样，那就合并成一个（所以一直合并的最终结果就是长度为2的树咯？） 
        if(gx!=gy)
        {
            //emm毕竟一般来说都是x吃y，所以x是y的上级祖宗，加之它是有方向的矢量，所以方便编程就是
            fa[gy]=gx;
            //d-1=实际捕食关系， 矢量相减就是食性距离，加3防止数字变负 
            eat[gy]=((d-1)+eat[x]-eat[y]+3)%3;
            continue;
        }
        //他们已经是一个祖宗了 
        else{
                if(d==1)//同类 
                {
                    if(eat[x]!=eat[y])//同类的食性居然不一样？！ 
                        {cnt++;continue;}
                }
                if(d==2)//x吃y，注意这是矢量加减（据称;滑稽），所以有方向 
                {
                    if((eat[y]-eat[x]+3)%3!=1)//x吃y但是之前说x与y是同类/被捕食的关系 
                    {
                        {cnt++;continue;}
                    }
                }
        }
    }
    printf("%d
",cnt);
    return 0;
}

一开始自作聪明，把get函数里return 的值直接写成temp，结果TLE 因为当x=fa[x]的时候，temp根本不会被赋值啊。

没有注释的清爽压行版本~❤

#include <iostream>
#include <algorithm>
#include <cmath>
#include <stdio.h>
#include <cstring>
#include <string>
#include <cstdlib>
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <map>
#include <list>
#include <iomanip>
 #include <fstream>

using namespace std;
int fa[50009],eat[50009];

int get(int x)
{
    int temp;
    if(x==fa[x])
        return x;    
    temp=fa[x];
    fa[x]=get(temp);
    eat[x]=(eat[x]+eat[temp])%3; 
    return fa[x];
}

int main()
{
    int n,k,cnt=0;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i){
        fa[i]=i;
        eat[i]=0;
    } 
    for(int i=1;i<=k;++i)
    {
        int d,x,y;
        scanf("%d%d%d",&d,&x,&y);
         if(x>n||y>n||(x==y&&d==2))    {cnt++;continue;}
        int gx=get(x),gy=get(y);
        if(gx!=gy)
        {
            fa[gy]=gx;
            eat[gy]=((d-1)+eat[x]-eat[y]+3)%3;    continue;
        }
        else{
            if(d==1&&(eat[x]!=eat[y]))    {cnt++;continue;}
            if(d==2&&((eat[y]-eat[x]+3)%3!=1) ) {cnt++;continue;}
        }
    }
    printf("%d
",cnt);
    return 0;
}