题目: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;
}