洛谷 P1955 [NOI2015]程序自动分析 题解

每日一题 day22 打卡

Analysis

离散化+并查集

先离散化所有的约束条件，再处理所有e=1的条件，将i的祖先和j的祖先合并到一个集合中；e=0时，如果i的祖先与j的祖先在同一个集合中，说明约束条件不成立，反之亦然。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define int long long
 #define maxn 1000000+10
 using namespace std;
 inline int read()
 {
     int x=;
     bool f=;
     char c=getchar();
     for(; !isdigit(c); c=getchar()) if(c=='-') f=;
     for(; isdigit(c); c=getchar()) x=(x<<)+(x<<)+c-'';
     if(f) return x;
     return -x;
 }
 inline void write(int x)
 {
     if(x<){putchar('-');x=-x;}
     if(x>)write(x/);
     putchar(x%+'');
 }
 int T,n,num,flag;
 struct node
 {
     int xi,xj,e;
 }lim[maxn];
 int book[maxn*],f[maxn];
 inline bool cmp(node x,node y)
 {
     return x.e>y.e;
 }
 inline int find(int x)
 {
     if(f[x]==x) return x;
     return f[x]=find(f[x]);
 }
 signed main()
 {
     T=read();
     while(T--)
     {
         num=;
         flag=;
         memset(f,,sizeof(f));
         memset(lim,,sizeof(lim));
         memset(book,,sizeof(book));
         n=read();
         for(int i=;i<=n;i++)
         {
             lim[i].xi=read();
             lim[i].xj=read();
             lim[i].e=read();
             book[++num]=lim[i].xi;
             book[++num]=lim[i].xj;
         }
         sort(book+,book+num+);
         int new_num=unique(book+,book+num+)-book;
         for(int i=;i<=n;i++)
         {
             lim[i].xi=lower_bound(book+,book+new_num+,lim[i].xi)-book;
             lim[i].xj=lower_bound(book+,book+new_num+,lim[i].xj)-book;
         }
         sort(lim+,lim+n+,cmp);
         for(int i=;i<=new_num;i++) f[i]=i;
         for(int i=;i<=n;i++)
         {
             if(lim[i].e==)
             {
                 int f1=find(lim[i].xi),f2=find(lim[i].xj);
                 if(f1!=f2) f[f1]=f2;
             }
             else if(lim[i].e==)
             {
                 int f1=find(lim[i].xi),f2=find(lim[i].xj);
                 if(f1==f2)
                 {
                     flag=;
                     printf("NO\n");
                     break;
                 }
             }
         }
         if(flag==) printf("YES\n");
     }
     return ;
 }

请各位大佬斧正（反正我不认识斧正是什么意思）