codeforces 1131D-Gourmet choice

传送门：QAQQAQ

题意：有两个数组，一个数组有n个数，另一个数组有m个数。s[i][j]表示第一个数组第i个数和第二个数组第j个数的大小关系，要求构造出一种方案，使条件成立。

先考虑没有等于号的情况，那么就是裸的拓扑排序，现在多了一个等于号，就先进行并查集再拓扑就行了。

•注意点：1.在进行拓扑排序时一定要全部用father[x]作为节点

                2.一定要全部并查集做完以后再连边，否则可能刚连完father[x]就变了

 #include<bits/stdc++.h>
 using namespace std;
 const int N=;
 typedef pair<int,int> pii;

 int p[N],a[N];
 int n,m,judge=;
 int t[N],ans[N],b[N];
 char s[][];
 vector <int> v[];

 int f(int x)
 {
     if(p[x]==x) return x;
     return p[x]=f(p[x]);
 }

 void unite(int x,int y)
 {
     int xx=f(x),yy=f(y);
     p[yy]=p[xx];
 }

 int main()
 {
     memset(b,,sizeof(b));
     memset(t,,sizeof(t));
     scanf("%d %d",&n,&m);
     for(int i=;i<=n+m;i++) p[i]=i;
     for(int i=;i<=n;i++)
     {
         scanf("%s",s[i]+);
         for(int j=;j<=m;j++)
         {
             if(s[i][j]=='=')
             {
                 unite(i,j+n);
             }
         }
     }
     for(int i=;i<=n;i++)
     {
         for(int j=;j<=m;j++)
         {
             if(s[i][j]=='>') v[f(j+n)].push_back(f(i)),t[f(i)]++;
             else if(s[i][j]=='<') v[f(i)].push_back(f(j+n)),t[f(j+n)]++;
         }
     }
     priority_queue<pii> q;
     for(int i=;i<=n+m;i++) if(f(i)==i) judge++;
     for(int i=;i<=n+m;i++)
     {
         if(!t[p[i]]&&!b[p[i]])
         {
             q.push(make_pair(p[i],));
             b[p[i]]=;
         }
     }
     int sum=,num=;
     while(!q.empty())
     {
         int now=q.top().first,w=q.top().second;
         q.pop(); sum++;
         ans[now]=w;
         for(int i=;i<v[now].size();i++)
         {
             int pos=v[now][i];
             t[pos]--;
             if(!t[pos])
             {
                 q.push(make_pair(pos,w+));
             }
         }
     }
     if(sum<judge)
     {
         puts("No");
         return ;
     }
     puts("Yes");
     for(int i=;i<=n;i++) printf("%d ",ans[p[i]]); puts("");
     for(int i=n+;i<=m+n;i++) printf("%d ",ans[p[i]]); puts("");
     return ;
 }