食物鏈(並查集)

梦曦缘發表於2024-11-13

題目:https://ac.nowcoder.com/acm/contest/22904/1024

思路:這道題網路上有很多思路,可以開三個並查集,可以使用帶權並查集,但是有一個大佬的思路是這樣的,將總結點的數量增加到3n個,把整個節點區域分為n,2*n,3*n三個部分,我們可以物種a的一個節點對應物種b的兩個節點,如果是同類,我們就把他們透過範圍對應起來(如a_n,a_2*n,a_3*n和b_n,b_2*n,b _3*n對應),如果是不是同類,我們讓這個節點指向下一個範圍的節點(如a_n,a_2*n,a_3*n和b_2*n,b_3*n,b _n對應),如此一來我們只需要判斷不同區域範圍的節點之間的關係即可

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 55000;
int a[3*maxn];
void init(int x){
    for(int i=0;i<3*x;i++){
        a[i] = i;
    }
}
int find(int x){
    if(a[x] == x) return x;
    return a[x] = find(a[x]);
}
void join(int x,int y){
    if(find(x) != find(y)) {
        a[find(x)] = find(y);
    }
}
int main(){
    int n,k;
    scanf("%d %d",&n,&k);
    init(n);
    int ans =0;
    while(k--){
        int d,x,y;
        scanf("%d %d %d",&d,&x,&y);
        if((x<1 || x>n || y<1 || y > n) || (d==2 && x==y)) {
            ans++;
            continue;
        }
        if(d==1) {
            if(find(x)==find(y+n) || find(x+n)==find(y)) ans++;
            else {
                join(x,y);
                join(x+n,y+n);
                join(x+2*n,y+2*n);
            }
        }
        else if(d==2){
            if(find(x)==find(y) || find(x+n)==find(y)) ans++;
            else {
                join(x,y+n);
                join(x+n,y+2*n);
                join(x+2*n,y);
            }
        }
    }
    printf("%d",ans);
    return 0;
}

相關文章