Gym - 101243F Vitamins(思维+并查集)

题意

有三种药丸，白色W>红色R>蓝色B，给你m个约束条件，问你n个药丸的颜色，不能确定颜色输出‘？’

题解

如果1<2<3，只要找到2就能确定1和3的颜色

如果2=4，只要确定一个就能确定另一个

处理的时候先把=用并查集处理一下

在处理<和>号

代码

 #include<bits/stdc++.h>
 using namespace std;

 const int maxn=;
 const int maxm=maxn*maxn/;
 int f[maxn];
 int u[maxm],v[maxm],op[maxm];
 vector<int>big[maxn],small[maxn];
 char s[];
 int find(int x)
 {
     return f[x]==x?x:f[x]=find(f[x]);
 }
 int main()
 {
     freopen("input.txt","r",stdin);
     freopen("output.txt","w",stdout);
     int n,m;
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++)f[i]=i;
     for(int i=;i<=m;i++)
     {
         scanf("%s",s);
         int l=strlen(s),sum=;
         for(int j=;j<=l;j++)
         {
             if(j==l)
             {
                 v[i]=sum;
                 continue;
             }
             if(s[j]>=''&&s[j]<='')
                 sum=sum*+s[j]-'';
             else
             {
                 if(s[j]=='>')op[i]=;
                 if(s[j]=='<')op[i]=;
                 if(s[j]=='=')op[i]=;
                 u[i]=sum,sum=;
             }
         }
         if(op[i]==)f[find(u[i])]=find(v[i]);
     }
     for(int i=;i<=m;i++)
         if(op[i]==)
         {
             int fu=find(u[i]),fv=find(v[i]);
             small[fu].push_back(fv);
             big[fv].push_back(fu);
         }
         else if(op[i]==)
         {
             int fu=find(u[i]),fv=find(v[i]);
             small[fv].push_back(fu);
             big[fu].push_back(fv);
         }
     char ans[maxn];
     memset(ans,'?',sizeof ans);
     for(int i=;i<=n;i++)
     {
         if(small[i].size()>&&big[i].size()>)
         {
             ans[i]='R';
             for(int j=;j<small[i].size();j++)
                 ans[small[i][j]]='B';
             for(int j=;j<big[i].size();j++)
                 ans[big[i][j]]='W';
         }
     }
     for(int i=;i<=n;i++)
         if(f[i]==i)
         {
             for(int j=;j<=n;j++)
             {
                 if(ans[j]=='?'&&find(j)==f[i])
                     ans[j]=ans[i];
             }
         }
     for(int i=;i<=n;i++)
         printf("%c",ans[i]);
     printf("\n");
     return ;
 }