La 4255 Guess

题目：Guess

思路：拓扑排序+并查集

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 10

int n;

int a[maxn+5][maxn+5];
int cnt[maxn+5];

int fa[maxn+5]= {0};	//并查集

vector<int> od;

int sum[maxn+5]= {0};	//前缀和

bool b[maxn+5]= {0};	//是否已被删除

int find(int x) {	//并查集的路径压缩
	if(x==fa[x]) return x;
	return fa[x]=find(fa[x]);
}

void print() {	//输出
	for(int i=0; i<od.size(); i++) {
		int k=10-i,x=find(od[i]);
		for(int j=0; j<=n; j++) {
			if(find(j)==x) {
				sum[j]=k;
			}
		}
	}
	printf("%d",sum[1]-sum[0]);
	for(int i=2; i<=n; i++) {
		printf(" %d",sum[i]-sum[i-1]);
	}
	printf("\n");
}

void readin() {	//读入
	scanf("%d",&n);
	
	char s[maxn+5][maxn+5]={0};
	for(int i=1; i<=n; i++) {
		for(int j=i; j<=n; j++) {
			char x;
			while(~(x=getchar())&&!(x=='+'||x=='-'||x=='0'));
			s[i][j]=x;
			if(x=='0') {	//注意0,+,-一定要分开写 
				b[j]=1,fa[j]=i-1;
			}
		}
	}
	for(int i=1; i<=n; i++) {
		for(int j=i; j<=n; j++) {
			if(s[i][j]=='+') {
				if(!a[find(i-1)][find(j)])	//注意不能重复加边
					a[find(i-1)][find(j)]=1,cnt[find(i-1)]++;
			} else if(s[i][j]=='-') {
				if(!a[find(j)][find(i-1)])
					a[find(j)][find(i-1)]=1,cnt[find(j)]++;
			}
		}
	}
}

void init() {	//初始化
	for(int i=0; i<=maxn; i++) fa[i]=i;
	od.clear();
	memset(a,0,sizeof(a));
	memset(cnt,0,sizeof(cnt));
	memset(b,0,sizeof(b));
	memset(sum,0,sizeof(sum));
}

void tp() {	//拓扑排序
	while(true) {
		vector<int> td;
		for(int i=0; i<=n; i++) {
			if(cnt[i]==0&&!b[i]) {
				td.push_back(i);
			}
		}
		if(!td.size()) return ;
		
		for(int i=0; i<td.size(); i++) {
			for(int j=0; j<=n; j++) {
				if(a[j][td[i]]) a[j][td[i]]=0,cnt[j]--;
			}
			b[td[i]]=1;
		}

		find(td[0]);
		for(int i=1; i<td.size(); i++) {
			fa[find(td[i])]=fa[td[0]];
		}

		od.push_back(td[0]);
	}
}

int main() {
	int T;
	scanf("%d",&T);

	while(T--) {
		init();
		readin();
		tp();
		print();
	}

	return 0;
}