HDU - 4626 Jinkeloid

题面

题意

每组数据给出一个字符串（仅包含前20个字符），多次询问，每次询问给出k（k<=5）个字符，问给出的字符串中有几个子串含给出的字符都是偶数个（包括0）。

做法

因为一共只有20种字符，所以考虑状压，但若直接暴力dp求出每种情况的子串数，复杂度为O（|s|*2^20），显然会T。
可以将问题先转换一下，先状压前缀，每个字符串都可以看做是从前缀中选出两个不同位置上的数Xor值。
因为除了这k个字符外，其余字符均不用考虑，所以我们之前就可以预处理出每个状态的子集中一共有几个前缀。
若某一位上的数字为0，则两个前缀上的该位数字要么都为0，要么都为1，因此我们可以O（2^k）暴力枚举Xor的两个前缀的k位中哪几位都是1，然后再用容斥求出这i位是1，其余（k-i）位是0的前缀对数，将（2^k）个答案相加即为答案。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define C ch=getchar()
#define N 100100
#define M 1050000
#define MX 1048576
using namespace std;

ll T,n,m,cb[M],cnt[M],now,ans;
char str[N],st[10],ch;

int main()
{
    ll i,j,k,t,tmp;
    for(i=0;i<MX;i++)
    {
        for(j=0;j<20;j++)
        {
            if((1 << j)&i) cb[i]++;
        }
    }
    for(i=0;i<MX;i++) cb[i]&=1;
    cin>>T;
    while(T--)
    {
        memset(cnt,0,sizeof(cnt));
        scanf("%s",str+1);
        n=strlen(str+1);
        now=0;
        cnt[0]=1;
        for(i=1;i<=n;i++)
        {
            now^=(1 << (str[i]-'a'));
            cnt[now]++;
        }
        for(i=0;i<20;i++)
        {
            for(j=0;j<MX;j++)
            {
                if(j&(1 << i)) continue;
                cnt[j|(1 << i)]+=cnt[j];
            }
        }
        scanf("%lld",&m);
        while(m--)
        {
            ans=t=0;
            scanf("%lld",&k);
            for(i=1;i<=k;i++)
            {
                for(C;ch<'a';C);
                st[i]=ch;
                t|=(1 << (st[i]-'a'));
            }
            for(i=0;i<(1 << k);i++)
            {
                tmp=now=0;
                for(j=1;j<=k;j++)
                {
                    if((1 << (j-1))&i) now|=(1 << (st[j]-'a'));
                }
                for(j=now;;j=(j-1)&now)
                {
                    if(cb[now^j])
                    {
                        tmp-=cnt[(MX-1)^t^j];
                    }
                    else
                    {
                        tmp+=cnt[(MX-1)^t^j];
                    }
                    if(!j) break;
                }
                ans+=tmp*(tmp-1)/2;
            }
            printf("%lld\n",ans);
        }
    }
}