食物鏈帶權並查集(POJ 1182)

動物王國中有三類動物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

將A記爲[x,y],B記爲[x+n,y+n],C記爲[x+2*n,y+2*n];所以在排入集合時

if(same(x,y+n) || same(x,y+n*2))
            {
                cnt++;
                continue;
            }
            Union(x,y);
            Union(x+n,y+n);
            Union(x+n*2,y+n*2);

又A吃B,B吃C,C吃A;

if(same(x,y)||(same(x,y+2*n)))
            {
                cnt++;
                continue;
            }
            Union(x,y+n);
            Union(x+n,y+2*n);
            Union(x+2*n,y);
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int maxn = 3*5e4+5;
int fa[maxn],n,m;
int findfa(int x)
{
    return x==fa[x]?x:fa[x] = findfa(fa[x]);
}
void Union(int a,int b)
{
    int x = findfa(a),y = findfa(b);
    if(x != y)
        fa[x] = y;
}
int same(int a,int b)
{
    return findfa(a) == findfa(b);
}

int main()
{
    scanf("%d%d",&n,&m);
    int d,x,y,cnt = 0;
    for(int i = 1; i < maxn; i++)
        fa[i] = i;
    for(int i = 0; i < m; i++)
    {
        scanf("%d%d%d",&d,&x,&y);
        if(x < 1 || y < 1 || x > n || y > n)
        {
            cnt++;
            continue;
        }
        if(d == 1)
        {
            if(same(x,y+n) || same(x,y+n*2))
            {
                cnt++;
                continue;
            }
            Union(x,y);
            Union(x+n,y+n);
            Union(x+n*2,y+n*2);
        }
        else
        {
            if(same(x,y)||(same(x,y+2*n)))
            {
                cnt++;
                continue;
            }
            Union(x,y+n);
            Union(x+n,y+2*n);
            Union(x+2*n,y);
        }
    }
    cout<<cnt<<endl;
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章