1. 程式人生 > >1182(帶權並查集)

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

程式碼:

#include<iostream>
#include<stdio.h>
using namespace std;

int fa[50005], dfa[50005];

int find(int x)		//找祖先
{
	if (fa[x] == x)return x;
	int f = fa[x];
	fa[x] = find(f), dfa[x] += dfa[f];
	return fa[x];
}

int main()
{
	int n, m, d, x, y, ans = 0;
	scanf("%d%d", &n, &m);
for (int i = 0; i <= n; i++)fa[i] = i, dfa[i] = 0; while (m--) { scanf("%d%d%d", &d, &x, &y); if (x > n || y > n) { ans++; continue; } int fx = find(x), fy = find(y); if (fx != fy)fa[fx] = fy, dfa[fx] = dfa[y] - dfa[x] + d - 1; else if (((dfa[x] - dfa[y]) %
3 + 3) % 3 != d - 1)ans++; } cout << ans << endl; return 0; }