bzoj4195: [Noi2015]程式自動分析(離散化+並查集)

Hanks_o發表於2018-03-15

題目傳送門

解法:
相等的可以用一個並查集維護在一起。
那麼當所有相等的都連好以後。
問所有不相等的。如果不相等的在同一個塊那麼說明有問題。。
離散化是顯然的啦

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct node {int s,id;}a[2100000],b[2100000];
struct trnode {int x,y,next;}e[2100000];int len,last[1100000];
void ins(int x,int y) {len++;e[len].x=x;e[len].y=y;e[len].next=last[x];last[x]=len;}
bool cmp(node n1,node n2) {return n1.s<n2.s;}int v[1100000];bool bo[1100000];
int fa[1100000];int findfa(int x) {if(x!=fa[x])fa[x]=findfa(fa[x]);return fa[x];}
int main() {
    int T;scanf("%d",&T);
    while(T--) {
        int n;scanf("%d",&n);int t=0;
        for(int i=1;i<=n;i++) {
            t++;scanf("%d",&a[t].s);a[t].id=i;
            t++;scanf("%d",&a[t].s);a[t].id=i;
            scanf("%d",&v[i]);
        }sort(a+1,a+1+t,cmp);
        int tt=0;memset(bo,false,sizeof(bo));
        for(int i=1;i<=t;i++) {
            if(a[i].s!=a[i-1].s)tt++;
            if(bo[a[i].id]==false) {bo[a[i].id]=true;b[a[i].id*2-1].s=tt;}
            else b[a[i].id*2].s=tt;
        }for(int i=1;i<=tt;i++)fa[i]=i;
        len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++) {
            int x=b[i*2-1].s,y=b[i*2].s;
            if(v[i]==1) {
                int xx=findfa(x),yy=findfa(y);
                if(xx!=yy)fa[xx]=yy;
            }
        }
        bool bk=true;
        for(int i=1;i<=n;i++) {
            int x=b[i*2-1].s,y=b[i*2].s;
            if(v[i]==0) {
                int xx=findfa(x),yy=findfa(y);
                if(xx==yy){bk=false;break;}
            }
        }
        if(bk==false)printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}

相關文章