bzoj 3160: 万径人踪灭【FFT+manacher】

时间:2021-12-05 12:12:26

考虑正难则反,我们计算所有对称子序列个数,再减去连续的

这里减去连续的很简单,manacher即可

然后考虑总的,注意到关于一个中心对称的两点下标和相同(这样也能包含以空位为对称中心的方案),所以设f[i]为下标和为i的对称中心一共有多少对相同字符,这样总答案就是\( \sum_{i=0}{2*n-2}2{f[i]}-1 \)(减掉的1是减掉空集)

然后考虑f怎么求,\( f[i]=((\sum_{j=0}^{i-1}s[j]s[i-j])+1)/2 \),除2是因为每一对都被算了两遍

暴力是不行的,但是这个(j,i-j)看着非常卷积,所以考虑怎么用FFT优化一下,把a字符和b字符分开算,以算a字符为例,做一个数组a[i]=(s[i]'a')?1:0,然后多项式是\( \sum_{j=0}^{i-1}a[j]*a[i-j]) \),这样就只算了对称且两点都是a的方案数,b同理

然后减掉manacher结果即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int N=500005,mod=1e9+7;
int n,f[N],bt,lm,re[N];
long long ans;
char c[N],s[N];
struct cd
{
double a,b;
cd(double A=0,double B=0)
{
a=A,b=B;
}
cd operator + (const cd &x) const
{
return cd(a+x.a,b+x.b);
}
cd operator - (const cd &x) const
{
return cd(a-x.a,b-x.b);
}
cd operator * (const cd &x) const
{
return cd(a*x.a-b*x.b,a*x.b+b*x.a);
}
}a[N],b[N];
void dft(cd a[],int f)
{
for(int i=0;i<lm;i++)
if(i<re[i])
swap(a[i],a[re[i]]);
for(int i=1;i<lm;i<<=1)
{
cd wi=cd(cos(M_PI/i),f*sin(M_PI/i));
for(int k=0;k<lm;k+=(i<<1))
{
cd w=cd(1,0),x,y;
for(int j=0;j<i;j++)
{
x=a[j+k],y=w*a[i+j+k];
a[j+k]=x+y,a[i+j+k]=x-y;
w=w*wi;
}
}
}
if(f==-1)
for(int i=0;i<lm;i++)
a[i].a/=lm;
}
long long ksm(long long a,long long b)
{
long long r=1;
while(b)
{
if(b&1)
r=r*a%mod;
a=a*a%mod;
b>>=1;
}
return r;
}
int main()
{
scanf("%s",c);
n=strlen(c);
for(bt=0;(1<<bt)<=2*n;bt++);
lm=(1<<bt);
for(int i=0;i<lm;i++)
re[i]=(re[i>>1]>>1)|((i&1)<<(bt-1));
for(int i=0;i<lm;i++)
a[i].a=(c[i]=='a')?1:0,a[i].b=0;
dft(a,1);
for(int i=0;i<lm;i++)
b[i]=a[i]*a[i];
for(int i=0;i<lm;i++)
a[i].a=(c[i]=='b')?1:0,a[i].b=0;
dft(a,1);
for(int i=0;i<lm;i++)
b[i]=b[i]+a[i]*a[i];
dft(b,-1);
for(int i=0;i<=2*(n-1);i++)
ans=(ans+ksm(2,((int)(b[i].a+0.5)+1)/2)-1)%mod;//cerr<<ans<<endl;
for(int i=0;i<n;i++)
s[(i+1)*2]=c[i],s[(i+1)*2+1]='#';
s[0]='$',s[1]='#',s[2*n+2]='&';
for(int i=1,mx=0,w;i<2*n+1;i++)
{
if(i<mx)
f[i]=min(f[2*w-i],mx-i);
else
f[i]=1;
for(;s[i-f[i]]==s[i+f[i]];f[i]++);
if(i+f[i]>mx)
mx=i+f[i],w=i;
ans=(ans-f[i]/2+mod)%mod;
}
printf("%lld\n",(ans%mod+mod)%mod);
return 0;
}