UVALive 4255-Guess-拓扑排序

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=21664

题意：对于一个序列a1，a2...an，我们可以计算出一个符号矩阵S，其中Sij为ai+..+aj的正负号或0，代表下标i累加到j的元素和为正负或0。给出符号矩阵，要求输出一个对应的序列。

对于Sij，我们可以看成前缀和SUM[J]-SUM[I-1]

那么每个符号就对应了 一条有向边，图的节点共n+1个，为0 1 2...n

我们对其拓扑排序，便可以得到一个满足输入中所有+ -符号 的序列

但是还不能满足 0的要求  符号为0即是 sum[j]==sum[i-1]，因此我们直接让这两个点的值相同即可

因此我们用并查集把相等的元素都指向其祖先

最后输出的时候 让sum[i] =sum[find(i)]即可

【但是注意并查集中，应该取序号小的为祖先，因为我的拓扑排序中，没有先后关系的节点是按 序号排列的 】

例如 X1 2 3 X2 

这种情况下，如果让x1=x2=4，那么会导致拓扑序出错，

如果是x2=x1=1，则不会影响拓扑序。。。

最后前缀和相减即可得到答案数组

 
// Rujia Liu
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
 
int n;
int du[25];
char tm[20][20];
int mp[20][20];
vector<int> topo;
void top()
{
	topo.clear();
	int i,j;
	int cun=n+1;
	while(cun)
	{
		for (i=0;i<=n;i++)
		{
			if (du[i]) continue;
			else break;
		}
		topo.push_back(i);
		du[i]=-1;cun--;
		for (j=0;j<=n;j++)
		{ 
			if (mp[i][j])
				du[j]--;
		}
	}
}  
int fa[25];
int find(int x)
{return x==fa[x]?x:(fa[x]=find(fa[x]));}
char ss[25*25];
int main(){
	int t;
	int i,j;
	scanf("%d",&t);
	while(t--)
	{
		memset(du,0,sizeof du);
		memset(mp,0,sizeof mp);
		memset(tm,0,sizeof tm);

		scanf("%d",&n);
		scanf("%s",ss); 
		int idx=0;
		 
		for (i=1;i<=n;i++)   fa[i]=i;

		for (i=1;i<=n;i++)  
			for (j=1;j<=n;j++) 
			{
				if (i<=j)  
				{
					tm[i][j]=ss[idx++];   
					if (tm[i][j]=='0')
						fa[j]=find(i-1);		//优先指向序号小的为祖先
				}
			}  
				for (i=1;i<=n;i++) 
					for (j=1;j<=n;j++) 
					{
						if (tm[i][j]=='+')
							mp[i-1][j]=1,du[j]++;
						if (tm[i][j]=='-')
							mp[j][i-1]=1,du[i-1]++;
					} 
					top();
					int sum[25]; 
					int ok=0; 
					for (i=0;i<=n;i++)
						sum[topo[i]]=ok++; 
					for (i=1;i<=n;i++)
					{   
						sum[i]=sum[find(i)];	//值与祖先相同
						if (i>1) printf(" ");
						printf("%d",sum[i]-sum[i-1]);
					}
					printf("\n");
	}
	
	
	return 0;
}