文章标题

题意:给一个字符串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;
}