bzoj 3160 万径人踪灭——FFT

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3160

似乎理解加深了。

用卷积算相同的位置；先把 a 赋成1、 b 赋成0，卷积一遍；再把 a 赋成0、 b 赋成1，卷积一遍；两个加起来就有了每个位置的值，它表示以该位置/2（/2的位置可以是裂缝）为对称轴的回文位置个数。

然后用马拉车把连续区间的情况去掉。

注意一下单个元素也要算上，因为有那种奇数的；马拉车里别忘了把单个元素减去。

因为FFT的两个数组是一样的，所以FFT一次，然后自己乘自己就行了。

虽然调了很久但其实也没什么要注意的。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
#define db double
using namespace std;
const int N=1e5+,M=N<<,mod=1e9+;
const db pi=acos(-);
int n,r[M],len,ans,bin[N];
char ch[N],tc[N<<];
struct cpl{db x,y;}a[M],b[M],I;
cpl operator+ (cpl a,cpl b){return (cpl){a.x+b.x,a.y+b.y};}
cpl operator- (cpl a,cpl b){return (cpl){a.x-b.x,a.y-b.y};}
cpl operator* (cpl a,cpl b){return (cpl){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};}
void upd(int &x){x>=mod?x-=mod:;x<?x+=mod:;}
void fft(cpl *a,bool fx)
{
  for(int i=;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=;R<=len;R<<=)//<<=1
    {
      int m=R>>;
      cpl Wn=(cpl){ cos(pi/m),fx?-sin(pi/m):sin(pi/m) };
      for(int i=;i<len;i+=R)
    {
      cpl w=I;
      for(int j=;j<m;j++,w=w*Wn)//w=w*Wn
        {
          cpl tmp=w*a[i+m+j];
          a[i+m+j]=a[i+j]-tmp;
          a[i+j]=a[i+j]+tmp;
        }
    }
    }
}
void manachar()
{
  int m=(n<<)-;
  for(int i=,j=;j<n;i+=,j++)
    tc[i]=ch[j],tc[i+]=',';
  int id=;
  r[]=;ans--;
  for(int i=;i<m;i++)
    {
      r[i]=;
      if(id+r[id]>i)
    {
      r[i]=min(id+r[id]-i,r[(id<<)-i]);
    }
      if(id+r[id]<=i+r[i])
    {
      for(;i+r[i]<m&&i-r[i]>=&&tc[i+r[i]]==tc[i-r[i]];r[i]++);
      r[i]--; id=i;
    }
      int tmp=(r[i]+(tc[i]==','))>>;//don't change r[i]!!!!!!
      ans-=tmp+(tc[i]!=','); upd(ans);
    }
}
int main()
{
  I.x=;
  scanf("%s",ch);n=strlen(ch);
  bin[]=;
  for(int i=,m=n+>>;i<=m;i++)bin[i]=bin[i-]<<,upd(bin[i]);
  len=;
  for(;len<=n<<;len<<=);
  for(int i=;i<len;i++) r[i]=(r[i>>]>>)+((i&)?len>>:);
  for(int i=;i<n;i++)
    if(ch[i]=='a')a[i].x=;
  fft(a,);
  for(int i=;i<len;i++)a[i]=a[i]*a[i];
  fft(a,);
  for(int i=;i<n;i++)
    if(ch[i]=='b')b[i].x=;
  fft(b,);
  for(int i=;i<len;i++)b[i]=b[i]*b[i];
  fft(b,);
  for(int i=;i<len;i++)//len
    {
      a[i].x=(int(a[i].x/len+0.5)+)>>;
      b[i].x=(int(b[i].x/len+0.5)+)>>;
      a[i].x+=b[i].x;
      ans+=bin[(int)a[i].x]-; upd(ans);
    }
  manachar();
  printf("%d\n",ans);
  return ;
}