poj1703

关注

原创最新推荐文章于 2025-03-31 19:31:39 发布 442 阅读 AI 写同款· GEO 优化 ›

poj 1703(并查集) 题意：有n个人和两个阵营，给出两种操作，A a b ，询问a和b的是否在同一个阵营，如果有人未加入任何阵营输入不确定。D a b，让a和b成为不同阵营的。题解：两种做法。一种是把并查集扩大一倍，先给每个人一个假定的敌人，然后有D操作时把敌人加入到假定敌人的集合中，A操作判断a和b是否在同一个集合，如果不在判断对方是否在自己假定敌人的集合中，在就是敌人，否则是不确定。#include <std 阅读详情

链接：点击打开链接

题意：有N个罪犯，M条关系有两种可能，分别为"D[a][b]"代表a，b为不同团体，"A [a] [b]"则输出相应的关系

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int par[1000005],rank[1000005];
int T,N,M;
void init(int n){
    int i;
    for(i=0;i<=2*n;i++){
        par[i]=i;
        rank[i]=0;
    }
}
int find(int x){
    if(par[x]==x)
    return par[x];
    return par[x]=find(par[x]);
}
void unite(int x,int y){
    x=find(x);
    y=find(y);
    if(x==y)
    return;
    if(rank[x]<rank[y])
    par[x]=y;
    else{
        par[y]=x;
        if(rank[x]==rank[y])
        rank[x]++;
    }
}
bool same(int x,int y){
    return find(x)==find(y);
}                                   //并查集模板
int main(){
    int i,j,x,y;
    char c;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&N,&M);
        init(2*N);
        for(i=0;i<M;i++){
            scanf(" %c",&c);
            scanf("%d%d",&x,&y);
            if(c=='D'){             //用x表示属于x属于第一个团体，x+N属于第二个团体，用两个人的关系
                unite(x,y+N);       //维护每个集合，每个集合的意义在于集合中所有的事情一定同时发生
                unite(y,x+N);       //也就是所有人都符合逻辑关系，而不一定是所有人都是同一个团体
            }
            else{
                if(same(x,y))
                puts("In the same gang.");
                else if(same(x,y+N)||same(x+N,y))
                puts("In different gangs.");
                else
                puts("Not sure yet.");
            }
        }
    }
    return 0;
}

收录于