POJ2492 並查集的拓展

前一段時間學了簡單的並查集，網上逛技術部落格，看到並查集的拓展，在原有並查集的基礎上，加入集合內部元素和其父節點之間的關係，這樣的拓展，可以解決更多問題

題目連結：

http://poj.org/problem?id=2492

題目大意：

輸入n個bug，bug之間有interaction，當前假設異性之間才interaction，但是需要驗證，給定這些interaction對，判定是否滿足假設

如果男<->女和女<->男，滿足條件，如果存在男<->男或者女<->女，則假設不滿足

方法：

採用並查集拓展，加入relations[]數組，其中relations[x]表示x和父節點關係，relations[x] = 0表示x和父節點關係為同性，relations[x] = 1表示x和父節點關係為異性

代碼如下：

#include <cstdio>#include <cstring>#include <cstdlib>using namespace std;#define MAX 5005int parent[MAX];//記錄父節點int relations[MAX];//relations[x]記錄x和父節點之間的關係//其中relations[x] = 0表示x和父節點關係為同性， relations[x] = 1表示x和父節點關係為異性//集合尋找int findSet(int x){int temp = parent[x];if(temp == x)return x;parent[x] = findSet(temp);relations[x] = (relations[x] + relations[temp]) % 2;//根據老的父節點和新父節點關係，修改relations[x]值return parent[x];}//合并集合void unionSet(int root1, int root2){int x = findSet(root1);int y = findSet(root2);if(x == y)return ;parent[x] = y;relations[x] = ((relations[root1] - relations[root2] + 1) % 2);return ;}//初始化資料void init(void){memset(relations, 0, sizeof(relations));for(int i = 0; i < MAX; i++)parent[i] = i;return ;}int main(){int temp1, temp2;int n, m;int num;int count = 1;scanf("%d", &num);while(num--){scanf("%d%d", &n, &m);init();int flag = 1;for(int i = 0; i < m; i++){scanf("%d%d", &temp1, &temp2);if(findSet(temp1) == findSet(temp2)){if(relations[temp1] != (relations[temp2] + 1) % 2)//如果不滿足異性關係，有矛盾flag = 0;}else{unionSet(temp1, temp2);}}if(flag){printf("Scenario #%d:\nNo suspicious bugs found!\n\n", count++);}else{printf("Scenario #%d:\nSuspicious bugs found!\n\n", count++);}}return 0;}