1. 程式人生 > >POJ 1182[並查集]

POJ 1182[並查集]

using ont ini printf %d sch 狀態 ios border

動物王國中有三類動物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[並查集]