10158 War 並查集升級版
阿新 • • 發佈:2019-01-01
題目大意:題目好長,就不細說了
解題思路:解法1:設定兩個陣列,一個數組是並查集的基本陣列p[maxn],另一個數組r[maxn]表示點和根節點的關係,0表示和根節點是朋友,1表示和根結點是敵人,那麼在並查集的時候就可以順便設定以下r陣列了,這裡用到的是向量思想,什麼是向量思想,下面會給連結,這裡就不解釋了
解法2:給每個人設定一個永遠的敵人,他和他永遠的敵人不在同一個集合,然後只要結合這四個點進行判斷即可
解法1程式碼:
#include<stdio.h> #include<string.h> #define MAXN 10010 int N, p[MAXN], delta[MAXN]; int find(int x) { if(p[x] == x) return x; int tx = find(p[x]); delta[x] =(delta[x] + delta[p[x]]) % 2; p[x] = tx; return tx; } void init() { int i, x, y, tx, ty, flag; for(i = 0; i < N; i ++) { p[i] = i; delta[i] = 0; } for(;;) { scanf("%d%d%d", &flag, &x, &y); if(!x && !y && !flag) break; tx = find(x); ty = find(y); if(flag == 3) { if(tx != ty || delta[x] != delta[y]) printf("0\n"); else printf("1\n"); } else if(flag == 4) { if(tx != ty || delta[x] == delta[y]) printf("0\n"); else printf("1\n"); } else if(flag == 1) { if(tx == ty) { if(delta[x] != delta[y]) printf("-1\n"); } else { p[tx] = ty; delta[tx] = (delta[x] - delta[y] + 2) % 2; } } else if(flag == 2) { if(tx == ty) { if(delta[x] == delta[y]) printf("-1\n"); } else { p[tx] = ty; delta[tx] = (delta[x] - delta[y] + 1) % 2; } } } } int main() { while(scanf("%d", &N) == 1) { init(); } return 0; }
解法2程式碼:
#include<cstdio> #include<cstring> #define maxn 20020 int p[maxn],x1,x2,y1,y2,N; int find(int x) { return x == p[x] ? p[x]:p[x] = find(p[x]); } int main() { scanf("%d",&N); for(int i = 0; i <= 2 * N; i++) p[i] = i; int c,x,y; while(scanf("%d%d%d",&c,&x,&y) != EOF && c + x + y) { x1 = find(x); y1 = find(x + N); x2 = find(y); y2 = find(y + N); if(c == 1) { //如果x和y的敵人是朋友 if(x1 == y2) { printf("-1\n"); continue; } //如果x和y是朋友,那麼x的敵人就是y的敵人的朋友,y的敵人就是x的敵人的朋友 p[x1] = x2; p[y1] = y2; } if(c == 2) { //如果x和y是朋友 if(x1 == x2) { printf("-1\n"); continue; } //如果x和y是敵人,那麼x的敵人就是y的朋友,y的敵人就是x的朋友 p[x1] = y2; p[x2] = y1; } if(c == 3) { if(x1 == x2) printf("1\n"); else printf("0\n"); } if(c == 4) { if(x1 == y2) printf("1\n"); else printf("0\n"); } } return 0; }