D. Gourmet choice

 http://codeforces.com/contest/1131/problem/D

题意:给n*m的字符矩阵,表示两组数分别由n个数组成的数组A和m个数组成的数组B,aij是‘>’表示A中第i个数比B中第j个数大,'<'则相反,‘=’则两数相等;问是否有两数组能用字符矩阵表示,有则输出两数组中最大的数最小的数组;

思路:并查集+拓扑排序;把相等的用并查集缩点,再把小的向大连边;

#include<algorithm>
#include<set>
#include<vector>
#include<queue>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<set>
#include<vector>
#include<queue>
#include<cmath>
#include<cstring>
#include<sstream>
#include<cstdio>
#include<ctime>
#include<map>
#include<stack>
#include<string>
using namespace std;

#define sfi(i) scanf("%d",&i)
#define pri(i) printf("%d\n",i)
#define sff(i) scanf("%lf",&i)
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define INF 0x3f3f3f3f
#define eps 1e-6
#define PI acos(-1)
#define lowbit(x) ((x)&(-x))
#define zero(x) (((x)>0?(x):-(x))<eps)
#define fl() printf("flag\n")
ll gcd(ll a,ll b){while(b^=a^=b^=a%=b);return a;}
const int maxn=2e3+9;
const int mod=1e9+7;

template <class T>
inline void sc(T &ret)
{
    char c;
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0' && c <= '9')
    {
        ret = ret * 10 + (c - '0'), c = getchar();
    }
}

int ans[maxn],p[maxn],in[maxn];
int Find(int x)
{
    return p[x]==x?x:p[x]=Find(p[x]);
}
char s[maxn][maxn];
queue<int>q;
vector<int>g[maxn];
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n+m+2;i++) p[i]=i;
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s[i]+1);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(s[i][j]=='=')
            {
                int jx=Find(i);//相等的进行缩点
                int jy=Find(n+j);
                p[jx]=jy;
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(s[i][j]=='=') continue;
            if(Find(i)==Find(j+n)) //与相等矛盾
            {
                cout<<"No"<<endl;
                return 0;
            }
            //小指向大
            if(s[i][j]=='>')
            {
                int jx=Find(i);
                int jy=Find(j+n);
                g[jy].push_back(jx);
            }
            else if(s[i][j]=='<')
            {
                int jx=Find(i);
                int jy=Find(j+n);
                g[jx].push_back(jy);
            }
        }
    }
    for(int i=1;i<=n+m;i++)
    {
        g[i].erase(  unique(g[i].begin(),g[i].end()) ,  g[i].end()  );//去掉重边
    }
    //拓扑排序
    for(int i=1;i<=n+m;i++)
    {
        for(int j=0;j<g[i].size();j++)
        {
            int v=g[i][j];
            int vv=Find(v);
            in[vv]++;
        }
    }
    for(int i=1;i<=n+m;i++) if(in[Find(i)]==0)q.push(i),ans[i]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i<g[u].size();i++)
        {
            int nex=g[u][i];
            in[nex]--;
            ans[nex]=max(ans[nex],ans[u]+1);
            if(in[nex]==0) q.push(nex);
        }

    }
    //环不行
    for(int i=1;i<=n+m;i++)
    {
        if(in[i]>0)
        {
            cout<<"No"<<endl;
            return 0;
        }
    }
    cout<<"Yes"<<endl;
    for(int i=1;i<=n;i++)
    {
        cout<<ans[Find(i)]<<" ";
    }
    cout<<endl;
    for(int i=1;i<=m;i++)
    {
        cout<<ans[Find(i+n)]<<" ";
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_39132605/article/details/87924783