Gourmet choice CodeForces - 1131D

codeforces.com/problemset/problem/1131/D

先用并查集处理等号 通过大于小于号建图 拓扑判环 因为只把入度为0的点作为起点 所以拓扑可以顺带求出最长路

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn=1e3+10;

struct node
{
	int v,next;
};

queue <int> que;
node edge[maxn*maxn];
char e[maxn][maxn];
int f[2*maxn],first[2*maxn],degree[2*maxn],dis[2*maxn];
int n,m,num;

void addedge(int u,int v)
{
	edge[num].v=v;
	edge[num].next=first[u];
	first[u]=num++;
}

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

void unite(int u,int v)
{
	int fu,fv;
	fu=getf(u),fv=getf(v);
	if(fu!=fv){
		f[fv]=fu;
	}
}

bool toposort()
{
	int i,u,v,cnt;
	cnt=0;
	for(i=1;i<=n+m;i++){
		if(f[i]==i&&degree[i]==0){
			que.push(i);
			dis[i]=1;
			num--;
		}
	}
	while(!que.empty()){
		u=que.front();
		que.pop();
		for(i=first[u];i!=-1;i=edge[i].next){
			v=edge[i].v;
			degree[v]--,dis[v]=max(dis[v],dis[u]+1);
			if(degree[v]==0){
				que.push(v);
				num--;
			}
		}
	}
	return num==0;
}

int main()
{
	int i,j,fi,fj,flag;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++){
		scanf("%s",e[i]+1);
	}
	for(i=1;i<=n+m;i++){
		f[i]=i;
	}
	for(i=1;i<=n;i++){
		for(j=1;j<=m;j++){
			if(e[i][j]=='='){
				unite(i,j+n);
			}
		}
	}
	memset(first,-1,sizeof(first));
	num=0,flag=1;
	for(i=1;i<=n;i++){
		for(j=1;j<=m;j++){
			if(e[i][j]=='=') continue;
			fi=getf(i),fj=getf(j+n);
			if(fi==fj){
				flag=0;
				break;
			}
			if(e[i][j]=='<'){
				addedge(fi,fj);
				degree[fj]++;
			}
			else{
				addedge(fj,fi);
				degree[fi]++;
			}
		}
		if(!flag) break;
	}
	if(!flag){
		printf("No\n");
		return 0;
	}
	num=0;
	for(i=1;i<=n+m;i++){
		if(f[i]==i) num++;
	}
	if(toposort()){
		printf("Yes\n");
		for(i=1;i<=n;i++){
			fi=getf(i);
			printf("%d",dis[fi]);
			if(i<n) printf(" ");
			else printf("\n");
		}
		for(i=n+1;i<=n+m;i++){
			fi=getf(i);
			printf("%d",dis[fi]);
			if(i<n+m) printf(" ");
			else printf("\n");
		}
	}
	else printf("No\n");
	
	return 0;
}