程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> poj 1182 帶權並查集

poj 1182 帶權並查集

編輯：C++入門知識

poj 1182 帶權並查集

食物鏈 Time Limit: 1000MS Memory Limit: 10000K Total Submissions: 45303 Accepted: 13213

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

Sample Output

Source

Noi 01

[Submit] [Go Back] [Status] [Discuss]

存

#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int maxn= 50005;
typedef long long LL;
int p[maxn],d[maxn],n,k;
void init(int n)
{
    for(int i=1;i<=n;i++){
        p[i]=i;
        d[i]=0;
    }
}
int findx(int x)
{
    if(p[x]==x)return x;
    int root=findx(p[x]);
    d[x]+=d[p[x]];
    d[x]%=3;
    return p[x]=root;
}
bool Union(int x,int y,int op)
{
    if(x>n||y>n)return false;
    int u=findx(x),v=findx(y);
    if(u!=v)
    {
        p[u]=v;
        d[u]=(d[y]+op-d[x]+3)%3;
        return true;
    }
    return d[x]==(d[y]+op)%3;
}
int main()
{
    scanf("%d%d",&n,&k);
    init(n);

    int x,y,d,ans=0;
    while(k--)
    {
        scanf("%d%d%d",&d,&x,&y);
        if(!Union(x,y,d-1))ans++;
    }

    printf("%d\n",ans);
    return 0;
}