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
使用並查集維護和查詢兩個查詢對象的狀態。
第一種狀態 x和y在同一集合中,那麽x和y都在A,x和y都在B,x和y都在C;
第二種狀態 x吃y,那麽x在A,y在B,或者x在B,y在C,或者x在C,y在A;
每種狀態都標記出來,如果出現矛盾,那麽就ans加一。
比如說,本次說x和y在同一陣營,但是查詢發現,x已經吃了y,或者y吃了x,那麽當然就不行。
還有就是並查集還是寫成函數的好,這樣代碼清爽。不然只寫一個find(int x),(絕對不止我一個人這麽做),代碼會很難看,邏輯或許看上去不是那麽清楚。
#include <stdio.h> #include <string.h> #include <iostream> using namespace std; const int N = 50000 + 5; int fa[3 * N]; int n, k, x, y;int init(int n) { for(int i = 0; i <= n; i++) fa[i] = i; } int find(int x) { return x == fa[x] ? x : fa[x] = find(fa[x]); } void unite(int x, int y) { int xx = find(x); int yy = find(y); if(xx != yy) { fa[xx] = yy; } } bool same(int x, int y) { return find(x) == find(y); } void solve() { scanf("%d%d", &n, &k); init(3*n); int ans = 0; while(k--) { int op; scanf("%d%d%d", &op, &x, &y); x--; y--; if(x < 0 || x >= n || y < 0 || y >= n) { ans++; continue; } if(op == 1) { if(same(x, y + n) || same(x, y + 2 * n)) { ans++; } else { unite(x, y); unite(x + n, y + n); unite(x + 2 * n, y + 2 * n); } } else { if(same(x, y) || same(x, y + 2 * n)) { ans++; } else { unite(x, y + n); unite(x + n, y + 2 * n); unite(x + 2 * n, y); } } } printf("%d\n", ans); } int main() { solve(); return 0; }
POJ 1182[並查集]