20178.27 万径人踪灭 思考记录

字符串的题也可以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);
}