bzoj3160 万径人踪灭

Description

Input

Output

Sample Input

Sample Output

HINT

可以用总方案数减去不合法的【连续的】方案数。
后者用manacher跑一遍求。
前者把a和b分别看成1求一遍卷积，就是关于每个位置对称的字符个数$x$，$2^x-1$就是答案。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define LL long long
const double pi=acos(-1);
const int mod=1000000007;
struct com
{
    double a,b;
    com operator + (const com &c) const
    {
        return (com){a+c.a,b+c.b};
    }
    com operator - (const com &c) const
    {
        return (com){a-c.a,b-c.b};
    }
    com operator * (const com &c) const
    {
        return (com){a*c.a-b*c.b,a*c.b+b*c.a};
    }
}a[800010],b[800010],p[800010];
char s[200010];
int l,t,n,r[800010],f[200010],g[800010];
LL p2[1600010];
int rev(int x)
{
    int i,ret=0;
    for (i=0;i<t;i++)
        ret|=((x>>i)&1)<<t-i-1;
    return ret;
}
void init()
{
    int i;
    scanf("%s",s);
    n=strlen(s)*2+1;
    for (i=n-1;i>=0;i--)
        s[i]=i&1?s[i/2]:'$';
    t=0,l=1;
    while (l<2*n-1) t++,l<<=1;
    p[0].a=1;
    p[1]=(com){cos(2*pi/l),sin(2*pi/l)};
    for (i=2;i<l;i++)
    {
        p[i]=p[i>>1]*p[i>>1];
        if (i&1) p[i]=p[i]*p[1];
    }
    for (i=0;i<l;i++)
        r[i]=rev(i);
    p2[0]=1;
    for (i=1;i<l;i++)
        p2[i]=p2[i-1]*2%mod;
}
LL mana()
{
    int i,mx=0,id=0;
    LL ret=0;
    for (i=0;i<n;i++)
    {
        if (mx<=i) f[i]=1;
        else f[i]=min(f[2*id-i],mx-i);
        while (i-f[i]>=0&&i+f[i]<n&&s[i+f[i]]==s[i-f[i]]) f[i]++;
        /*printf("%d:%d\n",i,f[i]);*/
        if (i+f[i]-1>mx)
        {
            mx=i+f[i]-1;
            id=i;
        }
    }
    for (i=0;i<n;i++)
        ret=(ret+f[i]/2)%mod;
    //printf("mana:%lld\n",ret);
    return ret;
}
void fft(com *a,int c)
{
    int i,j,k,w;
    com u,v;
    for (i=0;i<l;i++)
        if (r[i]>i)
            swap(a[i],a[r[i]]);
    for (i=1;i<l;i<<=1)
        for (j=0;j<l;j+=(i<<1))
        {
            w=0;
            for (k=j;k<j+i;k++)
            {
                u=a[k];
                v=a[k+i]*p[w];
                a[k]=u+v;
                a[k+i]=u-v;
                w+=c*(l/i>>1);
                if (w<0) w+=l;
            }
        }
}
LL all()
{
    int i;
    LL ret=0;
    /*for (i=0;i<l;i++)
        a[i]=(com){s[i]=='a',0};
    fft(a,1);
    for (i=0;i<l;i++)
        a[i]=a[i]*a[i];
    fft(a,-1);
    for (i=0;i<l;i++)
        g[i]=int(a[i].a/l+0.5);
    for (i=0;i<l;i++)
        a[i]=(com){s[i]=='b',0};
    fft(a,1);
    for (i=0;i<l;i++)
        a[i]=a[i]*a[i];
    fft(a,-1);
    for (i=0;i<l;i++)
        g[i]+=int(a[i].a/l+0.5);*/
    memset(a,0,sizeof(a));
    for (i=0;i<n;i++)
        if (s[i]=='a')
            a[i/2].a=1;
    fft(a,1);
    for (i=0;i<l;i++)
        b[i]=a[i]*a[i];
    memset(a,0,sizeof(a));
    for (i=0;i<n;i++)
        if (s[i]=='b')
            a[i/2].a=1;
    fft(a,1);
    for (i=0;i<l;i++)
        b[i]=b[i]+a[i]*a[i];
    fft(b,-1);
    for (i=0;i<l;i++)
        g[i]=(LL)(b[i].a/l+0.5);
    for (i=0;i<l;i++)
        ret=(ret+p2[(g[i]+1)/2]-1+mod)%mod;
    //printf("all:%lld\n",ret);
    return ret;
}
int main()
{
    /*freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);*/
    init();
    printf("%lld\n",(all()-mana()+mod)%mod);
}