(CodeForces) Codeforces Round #541 (Div. 2) D. Gourmet choice (并查集+拓扑排序)

传送门

题目大意：第一天n个菜，第二天m个菜，一个n*m的矩阵代表他们之间的优劣关系，根据这个矩阵用数字来给每一个菜品打分，使得最大的数字最小。比如 Aij 是> 说明第一天的 i菜 比第二天的 j菜好。

解题思路：根据题目的描述，两道菜之间的优劣情况，我们可以连一条权值为1的单向边，而哪些没有指向他的点，那就是最小的数为1。从入度为0的点开始，一步步往里面走，我们自然可以想到拓扑排序，但这题目还有 = ，说明两个菜的优劣程度相同 那么考虑到，如果两个菜优劣程度相同，其实我们可以将他们认为是一个点(放到一个集合中)，他们满足的条件是共通的，所以我们可以用一个并查集去维护，我们把一个集合里的点都用一个点来代替，然后拓扑就行了，有环说明就矛盾。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
int fa[maxn],rk[maxn];
int n,m;
vector<int> eg[maxn];
void init() {
	for(int i=0; i<maxn; ++i) {
		fa[i]=i,rk[i]=1;
	}
}
int find(int x) {
	if(x==fa[x])	return fa[x];
	else return fa[x]=find(fa[x]);
}
void unite(int x,int y) {
	x=find(x),y=find(y);
	if(x!=y) {
		if(rk[x]>rk[y])	fa[y]=x;
		else {
			if(rk[x]==rk[y])	rk[y]++;
			fa[x]=y;
		}
	}
}
int in[maxn];
vector<int> num;
int ans[maxn];
void solve() { //拓扑排序
	int tp,sz=num.size(),cnt=sz;
	for(int i=0; i<sz; ++i) {
		for(int j=0; j<(int)eg[num[i]].size(); ++j) {
			tp=eg[num[i]][j];
			in[tp]++;
		}
	}
	queue<int> q;
	for(int i=0; i<sz; ++i) {
		if(!in[num[i]])	q.push(num[i]),ans[num[i]]=1;
	}
	int to;
	while(!q.empty()) {
		tp=q.front(),q.pop();
		cnt--;
		for(int i=0; i<(int)eg[tp].size(); ++i) {
			to=eg[tp][i];
			in[to]--;
			ans[to]=max(ans[to],ans[tp]+1); //更新ans
			if(!in[to]) {
				q.push(to);
			}
		}
	}
	if(cnt)	cout<<"No"<<endl; //说明有环，矛盾
	else {
		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)]<<" ";
		}
		cout<<endl;
	}
}
int main() {
	std::ios::sync_with_stdio(0);
	init();
	cin>>n>>m;
	char tp;
	for(int i=1; i<=n; ++i) {
		for(int j=1; j<=m; ++j) {
			cin>>tp;
			if(tp=='>')	eg[j+n].push_back(i);
			if(tp=='<')	eg[i].push_back(j+n);
			if(tp=='=')	unite(i,j+n);
		}
	}
	for(int i=1; i<=m+n; ++i) { //这一步就是用点来代表集合，这里感觉写繁琐了
		for(int j=0; j<(int)eg[i].size(); ++j) { //将指出去的边都指向集合的父亲
			int p=eg[i][j];
			if(find(p)!=p)	eg[i][j]=find(p);
		}
		if(find(i)!=i) { //用父亲代表这个集合
			eg[find(i)].insert(eg[find(i)].end(),eg[i].begin(),eg[i].end());
		} else	num.push_back(i); //现在的点放到num数组中
	}
	solve();
	return 0;
}