题意:给一个字符串s,并给定n个变量的长度
5
4 2 4 4 2
例如 s:1bad1=acbe
然后求满足此式子1bbaaaadddd1=aaaaccccbbee
的解有多少种(这些变量只能取0/1)
思路:
将该式子转换成1bbaaaadddd1aaaaccccbbee
先用pos记录a-z第一次出现的位置
然后用sum1记录每个连续串的首字符出现的位置
遍历一遍数组对每个非第一次出现的字符使用并查集
再从[1,len/2]与[len/2+1,len]进行一次并查集
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#define DEBUG 1
using namespace std;
typedef long long LL;
int len[500];
char s[1000];
char s1[10010];//把输入的字符串s按照字符个数全部合并(除了‘=’)得到s1
int sum1[500];//每个连续串的首字符出现的位置
int f[10010];
int pos[200];//每个第一个字符出现的位置
void init(int t)
{
for(int i=0; i<t; i++)
f[i]=i;
}
int Find(int t)
{
int x=t;
while(x!=f[x])
x=f[x];
while(t!=f[t])
{
int k=f[t];
f[t]=x;
t=k;
}
return x;
}
bool Union(int a,int b)
{
a=Find(a);
b=Find(b);
if(s1[a]=='0'&&s1[b]=='1'||(s1[a]=='1'&&s1[b]=='0')) return false;//如果出现0和1一起的话直接输出0
if(s1[b]<s1[a])//为了统一父亲...不然不知怎么处理0,1
f[a]=b;
else if(s1[b]==s1[a])
{
if(a<b)
f[b]=a;
else f[a]=b;
}
else f[b]=a;
return true;
}
int main()
{
#ifdef LOCAL
freopen("in.txt","r",stdin);
#endif
init(10000);
int n;
scanf("%d",&n);
for(int i='a'; i<'a'+n; i++)
scanf("%d",&len[i]);
len['0']=len['1']=1;
scanf("%s",s+1);
int slen=strlen(s+1),t=1;
for(int i=1,k=1; s[i]!=0; k++,i++)
{
if(s[i]=='=')
{
k--;
continue;
}
for(int j=0; j<len[s[i]]; t++,j++)
s1[t]=s[i];
if(k!=1&&s[i-1]!='=')
sum1[k]=sum1[k-1]+len[s[i-1]];
else sum1[k]=sum1[k-1]+1;
}
if(DEBUG)
for(int j=1; j<slen; j++)
printf("%d%c",sum1[j],j==slen-1?'\n':' ');
for(int j=1; j<t; j++)
if(!pos[s1[j]])
pos[s1[j]]=j;
for(int i=1; i<t; i+=len[s1[i]])
{
if(pos[s1[i]]!=i)
{
for(int j=0; j<len[s1[i]]; j++)
if(!Union(i+j,pos[s1[i]]+j))
{
puts("0");
return 0;
}
//printf("[%d %d] %d %d\n",i+j,pos[s1[i]]+j,f[3],f[2]);
}
}
for(int i=1; i<=t/2; i++)
if(!Union(t/2+i,i))
{
puts("0");
return 0;
}
//printf("[%d %d] %d %d\n",t/2+i,i,Find(t/2+i),f[i]);
LL ans=1;
for(int i=1; i<t; i++)
{
Find(i);
if(DEBUG)
printf("%d%c",f[i],i==t-1?'\n':' ');
if(s1[f[i]]!='0'&&s1[f[i]]!='1'&&f[i]==i)
ans*=2;
}
cout<<ans<<endl;
if(DEBUG)
{
puts(s1+1);
for(int i=1; i<=t/2; i++)
printf("%c",s1[i]);
puts("");
puts(s1+1+t/2);
}
return 0;
}