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=20001;
typedef pair<int,int> pii;

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

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