Codeforces Round #541 (Div. 2) Pro.D——并查集+拓扑排序

Round #541(Div.2) Pro.D Birthday

[题目链接]http://codeforces.com/contest/1131/problem/D
  总结：看到本题中的>和<条件，不难想到这道题要建图进行拓扑排序。但是由于‘=’的存在，不能简单地直接用dishes作为结点。思考后我们可以看到，‘=’连接的dish，在拓扑排序中是等价的，我们要判断是否存在拓扑排序，实际上要判断每个结点与它的等价类的拓扑排序。而对于等价类这一条件，我们可以转换为集合的形式：将等价的结点放在同一集合中，以集合为结点建图，再进行拓扑排序。
代码如下：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
#include <math.h>
#include <map>
#include <stdlib.h>
#include <ctype.h>
using namespace std;
#define ull unsigned long long
#define ll long long
const int M=1005,maxn=2005;
char a[M][M];
int pre[maxn],root[maxn],n,m;
struct Edge{
	int from,to;
	Edge(int u,int v):from(u),to(v){}
};
vector<Edge> edges;
vector<int> G[maxn];
int findRoot(int u){
	if(pre[u]==u) return u;
	return pre[u]=findRoot(pre[u]);
}
void AddEdge(int u,int v){
	edges.push_back(Edge(u,v));
	G[u].push_back(edges.size()-1);
}
void initial(void){
	scanf("%d%d",&n,&m);
	edges.clear();
	for(int i=1;i<=n+m;++i) G[i].clear();
	for(int i=1;i<=n+m;++i) pre[i]=i;
	for(int i=1;i<=n;++i){
		getchar();									
		for(int j=1;j<=m;++j){
			a[i][j]=getchar();				
			if(a[i][j]=='='){
				int x=findRoot(i),y=findRoot(n+j);
				if(x!=y) pre[x]=y;
			}
		}
	}
	for(int i=1;i<=n+m;++i) root[i]=findRoot(i);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=m;++j){
			if(a[i][j]=='>') AddEdge(root[i],root[j+n]);
			else if(a[i][j]=='<') AddEdge(root[j+n],root[i]);
		}
}
int val[maxn],visited[maxn];
bool dfs(int u){
	visited[u]=-1;val[u]=1;
	for(int i=0;i<G[u].size();++i){
		Edge& e=edges[G[u][i]];
		if(e.to==u) return false;
		if(visited[e.to]==-1) return false;
		if(!visited[e.to]&&!dfs(e.to)) return false;
		val[u]=max(val[u],val[e.to]+1);
	}
	visited[u]=1;
	return true;
}
bool TopoSort(void){
	memset(visited,0,sizeof(visited));
	for(int i=1;i<=n+m;++i)
		if(!visited[root[i]]&&!dfs(root[i]))
			return false;
	return true;
}
int main(void){
	initial();
	if(TopoSort()){
		puts("Yes");
		for(int i=1;i<=n;++i)
			printf("%s%d",i==1?"":" ",val[root[i]]);
		putchar('\n');
		for(int i=n+1;i<=n+m;++i)
			printf("%s%d",i==n+1?"":" ",val[root[i]]);
		putchar('\n');
	}
	else puts("No");
	return 0;
}