[SDOI2006]二进制方程 并查集

apio也有一个用并查集统计方案数的题，但这个题是弱化版。。而且网上竟然一份题解都没有

首先两个串能完全匹配，及说明相对应的两个位置有相等关系，在填0还是1的时候就只能填一次，所以就用并查集把相等的连到一起，统计答案时只统计fu【o】==o的部分

串中的0和1单独处理，特判即可

码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 10007
int fu[N],qsum[200],lena,lenb,za,zb,i,j,n,l[N],f1,f2,wza[N],wzb[N],ans[999999];
char cha[N],chb[N];
int find(int o)
{
	if(fu[o]!=o)fu[o]=find(fu[o]);
	return fu[o];
}
void c2()
{
	int i,jw=0;
	for(i=1;i<=ans[0];i++)
	{
int 	o=ans[i]*2+jw;
	ans[i]=o%10;
	jw=o/10;
    }
	while(jw)ans[++ans[0]]=jw%10,jw/=10;
}
int main()
{
	scanf("%d",&n);
	for(i=int('a');i<=int('a')+n-1;i++)
	scanf("%d",&l[i]),qsum[i]+=l[i-1]+qsum[i-1];
scanf("%s",cha);
scanf("%s",chb);
lena=strlen(cha);
lenb=strlen(chb);
l[int('1')]=1;l[int('0')]=1;
for(i=1;i<=10005;i++)fu[i]=i;
za=0;zb=0;
while(za<lena)
{
	wza[za]++;
	wzb[zb]++;
	if(cha[za]=='1')
	{
		f1=10001;
	}else if(cha[za]=='0')
	{
		f1=10002;
	}else f1=find(qsum[cha[za]]+wza[za]);
	
    if(chb[zb]=='1')
	{
		f2=10001;
	}else if(chb[zb]=='0')
	{
		f2=10002;
	}else f2=find(qsum[chb[zb]]+wzb[zb]);
	if(f1>10000&&f2>10000&&f1!=f2)
	{
		printf("0");
		return 0;
	}
	if(f1!=f2)
	{
	if(f1>10000)fu[f2]=f1;
	else fu[f1]=f2;
    }
	if(wza[za]==l[cha[za]])za++;
	if(wzb[zb]==l[chb[zb]])zb++;	
}
	ans[0]=1;
	ans[1]=1;
	for(i='a';i<='a'+n-1;i++)
	{
		for(j=1;j<=l[i];j++)
		{
			if(find(qsum[i]+j)==qsum[i]+j)c2();
		}
	}
	for(i=ans[0];i>=1;i--)
	printf("%d",ans[i]);
}