题目链接:http://codeforces.com/problemset/problem/1131/D
题目大意:有一个n*m的矩阵,里面填满比较符号,表示n个数字跟m个数字之间的关系,现在问能不能找到最大的数字最小的并且符合矩阵的所有n+m个数字。
思路:因为有’=’,所以先把所有相等的用并查集并为一个点,然后再把不相等的存入vector最后用拓扑排序解决。
代码:
#include<bits/stdc++.h>
using namespace std;
int fa[2010];
int du[2010];
vector<int>g[2010];
int gg[2010];
int L[2010];
int vis[2010];
int cc[2010];
int pro[2010];
int Find(int x)
{
int p,tmp;
p=x;
while(x!=fa[x])x=fa[x];
while(p!=x)
{
tmp=fa[x];
fa[x]=x;
p=tmp;
}
return x;
}
bool topsort(int n)
{
int xx=1;
int sum=0;
memset(du,0,sizeof du);
for(int i=1;i<=n;i++)
for(int j=0;j<g[gg[i]].size();j++)
du[g[gg[i]][j]]++;
int tot=0;
queue<int>Q;
for(int i=1;i<=n;i++)
if(!du[gg[i]])
{
Q.push(gg[i]);
pro[gg[i]]=xx;
sum++;
}
while(!Q.empty())
{
xx++;
int zz=0;
while(sum--)
{
int x=Q.front();
Q.pop();
L[++tot]=x;
for(int j=0;j<g[x].size();j++)
{
int t=g[x][j];
du[t]--;
if(!du[t])
{
Q.push(t);
zz++;
pro[t]=xx;
}
}
}
sum=zz;
}
if(tot==n)return 1;
return 0;
}
vector<int>vv[2010];
char s[2010][2010];
int main()
{
int n,m,i,j,fl=0,sum=0;
cin>>n>>m;
for(i=0;i<=n+m;i++)fa[i]=i;
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
{
cin>>s[i][j];
if(s[i][j]=='=')
{
int fx=Find(i),fy=Find(j+n);
if(fx!=fy)fa[fx]=fy;
}
}
for(i=1;i<=n+m;i++)
{
int fx=Find(i);
if(!vis[fx])
{
vis[fx]=1;
gg[++sum]=fx;
}
}
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
int fx=Find(i);
int fy=Find(j+n);
if(s[i][j]=='<')
{
if(fx==fy)fl=1;
g[fx].push_back(fy);
}
else if(s[i][j]=='>')
{
if(fx==fy)fl=1;
g[fy].push_back(fx);
}
}
}
if(fl==1)
{
cout<<"NO"<<endl;
return 0;
}
int C=topsort(sum);
if(C==0)
{
cout<<"NO"<<endl;
return 0;
}
cout<<"YES"<<endl;
for(i=1;i<=n;i++)
{
int fx=Find(i);
cout<<pro[fx]<<" ";
}
cout<<endl;
for(i=n+1;i<=m+n;i++)
{
int fx=Find(i);
cout<<pro[fx]<<" ";
}
}