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

题意

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

题解

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

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

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

在处理<和>号

代码

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 const int maxn=1005;
 5 const int maxm=maxn*maxn/2;
 6 int f[maxn];
 7 int u[maxm],v[maxm],op[maxm];
 8 vector<int>big[maxn],small[maxn];
 9 char s[50];
10 int find(int x)
11 {
12     return f[x]==x?x:f[x]=find(f[x]);
13 }
14 int main()
15 {
16     freopen("input.txt","r",stdin);
17     freopen("output.txt","w",stdout);
18     int n,m;
19     scanf("%d%d",&n,&m);
20     for(int i=1;i<=n;i++)f[i]=i;
21     for(int i=1;i<=m;i++)
22     {
23         scanf("%s",s);
24         int l=strlen(s),sum=0;
25         for(int j=0;j<=l;j++)
26         {
27             if(j==l)
28             {
29                 v[i]=sum;
30                 continue;
31             }
32             if(s[j]>='0'&&s[j]<='9')
33                 sum=sum*10+s[j]-'0';
34             else
35             {
36                 if(s[j]=='>')op[i]=1;
37                 if(s[j]=='<')op[i]=2;
38                 if(s[j]=='=')op[i]=3;
39                 u[i]=sum,sum=0;
40             }
41         }
42         if(op[i]==3)f[find(u[i])]=find(v[i]);
43     }
44     for(int i=1;i<=m;i++)
45         if(op[i]==1)
46         {
47             int fu=find(u[i]),fv=find(v[i]);
48             small[fu].push_back(fv);
49             big[fv].push_back(fu);
50         }
51         else if(op[i]==2)
52         {
53             int fu=find(u[i]),fv=find(v[i]);
54             small[fv].push_back(fu);
55             big[fu].push_back(fv);
56         }
57     char ans[maxn];
58     memset(ans,'?',sizeof ans);
59     for(int i=1;i<=n;i++)
60     {
61         if(small[i].size()>0&&big[i].size()>0)
62         {
63             ans[i]='R';
64             for(int j=0;j<small[i].size();j++)
65                 ans[small[i][j]]='B';
66             for(int j=0;j<big[i].size();j++)
67                 ans[big[i][j]]='W';
68         }
69     }
70     for(int i=1;i<=n;i++)
71         if(f[i]==i)
72         {
73             for(int j=1;j<=n;j++)
74             {
75                 if(ans[j]=='?'&&find(j)==f[i])
76                     ans[j]=ans[i];
77             }
78         }
79     for(int i=1;i<=n;i++)
80         printf("%c",ans[i]);
81     printf("\n");
82     return 0;
83 }

 

转载于:https://www.cnblogs.com/taozi1115402474/p/9668806.html