POJ 1182 食物鏈 加權並查集
阿新 • • 發佈:2019-01-22
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 2 3
1 1 3
2 3 1Sample Output
3
Source
Noi 01
據說這題擴充套件域並查集無腦過……可是……我並不會擴充套件域……所以打了加權並查集……
總之,不管加權還是擴充套件域,能A題的並查集就是好並查集!
程式碼
#include <iostream>
#include <cstdio>
using namespace std;
const int SZ = 150000;
int n, m, fa[SZ], rank[SZ];
int find(int x)
{
if(x == fa[x]) return x;
int f = fa[x];
fa[x] = find(fa[x]);
rank[x] = (rank[x] + rank[f]) % 3;
return fa[x];
}
bool judge(int x, int y, int d)
{
int a = find(x), b = find(y);
if(a == b)
{
if(d == 1 && rank[x] != rank[y]) return false;
if(d == 2)
{
if(rank[x] == 2 && rank[y] != 1) return false;
if(rank[x] == 1 && rank[y] != 0) return false;
if(rank[x] == 0 && rank[y] != 2) return false;
}
return true;
}
fa[a] = b;
if(d == 2)
rank[a] = (rank[y] - rank[x] + 3 + 1) % 3;
else
rank[a] = (rank[y] - rank[x] + 3) % 3;
return true;
}
int main()
{
scanf("%d%d", &n, &m);
int d, x, y;
for(int i = 1; i <= n; i ++)
fa[i] = i;
int cnt = 0;
for(int i = 1; i <= m; i ++)
{
scanf("%d%d%d", &d, &x, &y);
if(x > n || y > n || d == 2 && x == y)
{
cnt ++;
continue;
}
if(!judge(x, y, d))
cnt ++;
}
printf("%d", cnt);
return 0;
}