BZOJ3160: 万径人踪灭

BZOJ3160

题目看起来就很厉害的样子…实际上也很厉害啊！
题目其实是要求不连续的回文子序列个数，也就是回文子序列个数-连续的回文子序列个数。连续的回文子序列就是回文子串嘛，一遍$Manacher$就可以处理。
考虑回文子序列个数怎么求。
令$f[i]$表示以$i$为中心的对称点对个数。
那么答案就是$\sum(2^{f_i}-1)$
假设原串 “aba”
倍增串”$#a#b#a#”

第0个位置和第2个位置对倍增后第4个位子有贡献。
$f_{i}=\sum_{i=0}^{n-1}a_i*b_{n-i}$
$FFT$就可以啦

【代码】

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <complex>
#include <algorithm>
#include <cmath>
#define N 262145
#define INF 0x7fffffff
#define mod 1000000007
using namespace std;
typedef complex<double> C;
const double pi=acos(-1);
typedef long long ll;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

ll ans;
int n,m,mm,mx,id;
int p[200005],Bin[100005];
C a[N],b[N],f[N];
char s[100005],ss[200005];

void FFT(C *a,int n,int f)
{
    if(n==1) return;
    C wn(cos(2*pi/n),sin(2*pi*f/n)),w(1,0),t;
    C a0[n>>1],a1[n>>1];
    for(int i=0;i<n>>1;i++) a0[i]=a[i<<1],a1[i]=a[i<<1|1];
    FFT(a0,n>>1,f),FFT(a1,n>>1,f);
    for(int i=0;i<n>>1;i++,w*=wn) 
    {
        t=w*a1[i];
        a[i]=a0[i]+t;
        a[i+(n>>1)]=a0[i]-t;
    }
}

void Manacher()
{
    ss[0]='@',ss[1]='#';
    int len=mm+1<<1;
    for(int i=1;i<=mm+1;i++) ss[i<<1]=s[i-1],ss[i<<1|1]='#';
    for(int i=1;i<=len;i++)
    {
        p[i]=(i<mx)?min(mx-i,p[2*id-i]):1;
        while(ss[i+p[i]]==ss[i-p[i]]) p[i]++;
        ans-=p[i]>>1;ans%=mod;
        if(mx<i+p[i]) mx=i+p[i],id=i;
    }
}

int main() 
{
    scanf("%s",s);
    n=mm=strlen(s)-1;Bin[0]=1;
    for(int i=1;i<=n+1;i++) Bin[i]=(Bin[i-1]<<1)%mod;
    m=n<<1|1;for(n=1;n<=m;n<<=1);
    for(int i=0;i<=n;i++) a[i]=s[i]=='a',b[i]=s[i]=='b';
    FFT(a,n,1);FFT(b,n,1);
    for(int i=0;i<=n;i++) f[i]=a[i]*a[i]+b[i]*b[i];
    FFT(f,n,-1);
    for(int i=0;i<=n;i++)
    {
        int t=int(f[i].real()/n+0.5);
        ans+=Bin[t+1>>1]-1;ans%=mod;
    }
    Manacher();
    printf("%lld\n",(ans+mod)%mod);
    return 0;
}