1. 程式人生 > >BZOJ4195:[NOI2015]程序自動分析

BZOJ4195:[NOI2015]程序自動分析

代碼 puts max php spa fin har for unique

淺談並查集:https://www.cnblogs.com/AKMer/p/10360090.html

題目傳送門:https://lydsy.com/JudgeOnline/problem.php?id=4195

由於等於具有傳遞性,所以此題可以用並查集完美的解決。

把相等的元素放在一個集合裏,不等的判斷是不是不在一個集合裏即可。

時間復雜度:\(O(\alpha{n})\)

空間復雜度:\(O(n)\)

代碼如下:

#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn=1e5+5;

int n,cnt;
int fa[maxn<<1],tmp[maxn<<1];
int a[maxn],b[maxn],opt[maxn];

int read() {
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    return x*f;
}

int find(int x) {
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}

void merge(int a,int b) {
    a=find(a),b=find(b);
    if(a!=b)fa[a]=b;
}

int main() {
    int T=read();
    while(T--) {
        n=read();
        for(int i=1;i<=n;i++)
            a[i]=tmp[i]=read(),b[i]=tmp[i+n]=read(),opt[i]=read();
        sort(tmp+1,tmp+(n<<1)+1);
        cnt=unique(tmp+1,tmp+(n<<1)+1)-tmp-1;
        for(int i=1;i<=cnt;i++)fa[i]=i;
        for(int i=1;i<=n;i++) {
            a[i]=lower_bound(tmp+1,tmp+cnt+1,a[i])-tmp;
            b[i]=lower_bound(tmp+1,tmp+cnt+1,b[i])-tmp;
        }
        bool ans=1;
        for(int i=1;i<=n;i++)
            if(opt[i])merge(a[i],b[i]);
        for(int i=1;i<=n;i++)
            if(!opt[i]&&find(a[i])==find(b[i]))ans=0;
        if(ans)puts("YES");
        else puts("NO");
    }
    return 0;
}

BZOJ4195:[NOI2015]程序自動分析