字符串的题也可以fft、
只要能化成卷积的形式就都可以fft
比如这里的回文: 对一个位置:+1 -1 +2 -2 他们和相等,所以对a跑一边,对b跑一遍,就相当于建出n次多项式然后求系数
注意manacher不要写错,漏写最后一句
注意fft最后一句要写在外面
码:
#include<iostream>
#include<cstdio>
#include<complex>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 270005
#define P 1000000007
#define pi acos(-1)
typedef complex<double>E;
E a[N],b[N];
int er[N],l=0,cnt,n,i,j,k,p[N],m,r[N];
long long ans;
char ch[N],s[N];
void fft(E *a,int f)
{int i,j,k;
for(i=1;i<n;i++)if(i>r[i])swap(a[i],a[r[i]]);
for(i=1;i<n;i<<=1)
{
E wn(cos(pi/i),f*sin(pi/i));
for(j=0;j<n;j+=(i<<1))
{
E w(1,0);
for(k=0;k<i;k++,w*=wn)
{
E x=a[j+k],y=w*a[j+k+i];
// cout<<x<<endl;
a[j+k]=x+y,a[j+k+i]=x-y;
}
}
}
if(f==-1)for(i=0;i<n;i++)a[i]/=n;
}
void manach()
{ //cout<<cnt;
for(i=1;i<=cnt;i++)
{//if(i==cnt/4)cout<<"ee";
// if(i==cnt/2)cout<<"ee";
if(p[k]+k<i)
{
while(ch[i+p[i]]==ch[i-p[i]])++p[i];
}else
{
p[i]=min(p[k*2-i],p[k]+k-i);
while(ch[i+p[i]]==ch[i-p[i]])++p[i];
}
if(p[k]+k<p[i]+i)k=i;
}
for(i=1;i<=cnt;i++)
{
ans-=(p[i]-1)/2;
}
}
int main()
{
er[0]=1;
for(i=1;i<100000;i++)er[i]=er[i-1]*2%P;
scanf("%s",s);
int len=strlen(s);
ch[0]='$';
cnt=0;
for(i=0;i<len;i++)
{
ch[++cnt]='#';
ch[++cnt]=s[i];
}
ch[++cnt]='#';
manach();
for(i=0;i<len;i++)if(s[i]=='a')a[i]=1;
for(i=0;i<len;i++)if(s[i]=='b')b[i]=1;
n=len-1;
m=2*n;
for(n=1;n<=m;n<<=1)l++;
for(i=1;i<n;i++)r[i]=(r[i>>1]>>1)|((1&i)<<(l-1));
fft(a,1);for(i=0;i<n;i++)a[i]=(a[i]*a[i]);
fft(a,-1);
fft(b,1);for(i=0;i<n;i++)b[i]=(b[i]*b[i]);
fft(b,-1);
for(i=1;i<=m;i++)
{
if(i%2==1)
{
ans+=er[int(a[i].real()+0.1)/2+int(b[i].real()+0.1)/2]-1;
ans%=P;
}else
{
ans+=(er[int(a[i].real()+0.1)/2+int(b[i].real()+0.1)/2]-1)*2;
ans%=P;
}
}
printf("%lld",(ans+P)%P);
}