1. 程式人生 > >10158 War 並查集升級版

10158 War 並查集升級版

題目大意:題目好長,就不細說了

解題思路:解法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;
}