# BZOJ 3160: 万径人踪灭

FFT+Manacher.

## Code

/**************************************************************
Problem: 3160
User: BeiYu
Language: C++
Result: Accepted
Time:3868 ms
Memory:29616 kb
****************************************************************/

#include <bits/stdc++.h>
using namespace std;

#define mpr make_pair
#define rr first
#define ii second
typedef pair< double,double > Complex;
typedef long long LL;
const int N = 5e5+50;
const long long p = 1e9+7;
const double Pi = M_PI;

Complex operator + (const Complex &a,const Complex &b) {
return mpr(a.rr+b.rr,a.ii+b.ii);
}
Complex operator - (const Complex &a,const Complex &b) {
return mpr(a.rr-b.rr,a.ii-b.ii);
}
Complex operator * (const Complex &a,const Complex &b) {
return mpr(a.rr*b.rr-a.ii*b.ii,a.rr*b.ii+a.ii*b.rr);
}

int n,l;
LL ans;
int f[N],g[N];
char t[N],s[N];
Complex a[N],b[N],c[N];

LL Pow(LL a,LL b,LL r=1) { for(;b;b>>=1,a=a*a%p) if(b&1) r=r*a%p;return r; }
void Manacher(char *t) {
int ll=0;
for(int i=0;i<l;i++) s[ll++]='#',s[ll++]=t[i];
s[0]='\$',s[ll++]='@';

for(int i=0,j=0,mx=0;i<ll;i++) {
if(mx>i) f[i]=min(mx-i,f[j*2-i]);else f[i]=1;
while(i+f[i]<ll && i-f[i]>=0 && s[i+f[i]]==s[i-f[i]]) f[i]++;
if(i+f[i]>mx) mx=i+f[i],j=i;
}
}

void init(int x) {
for(n=1;n<x;n<<=1);n<<=1;
}
void Rev(Complex a[]) {
for(int i=0,j=0;i<n;i++) {
if(i>j) swap(a[i],a[j]);
for(int k=n>>1;(j^=k)<k;k>>=1);
}
}
void DFT(Complex a[],int r=1) {
Rev(a);
for(int i=2;i<=n;i<<=1) {
Complex wi=mpr(cos(2.0*Pi/i),r*sin(2.0*Pi/i));
for(int k=0;k<n;k+=i) {
Complex w=mpr(1.0,0.0);
for(int j=k;j<k+i/2;j++) {
Complex t1=a[j],t2=w*a[j+i/2];
a[j]=t1+t2,a[j+i/2]=t1-t2;
w=w*wi;
}
}
}
if(r==-1) for(int i=0;i<n;i++) a[i].rr/=n;
}
void FFT(Complex a[],Complex b[],Complex c[]) {
DFT(a),DFT(b);
for(int i=0;i<n;i++) c[i]=a[i]*b[i];
DFT(c,-1);
}

int main() {
scanf("%s",t);
l=strlen(t);
init(l);
for(int i=0;i<l;i++) a[i]=mpr(t[i]=='a',0),b[i]=a[i];
//  reverse(b,b+l);

//  for(int i=0;i<n;i++) cout<<(int)a[i].rr<<" ";cout<<endl;
//  for(int i=0;i<n;i++) cout<<(int)b[i].rr<<" ";cout<<endl;

FFT(a,b,c);

for(int i=0;i<n;i++) g[i]=(int)(c[i].rr+0.5);
//  for(int i=0;i<n;i++) cout<<(int)(c[i].rr+0.5)<<" ";cout<<endl;

memset(a,0,sizeof(a)),memset(b,0,sizeof(b));
for(int i=0;i<l;i++) a[i]=mpr(t[i]=='b',0),b[i]=a[i];
//  reverse(b,b+l);

//  for(int i=0;i<n;i++) cout<<(int)a[i].rr<<" ";cout<<endl;
//  for(int i=0;i<n;i++) cout<<(int)b[i].rr<<" ";cout<<endl;

FFT(a,b,c);

for(int i=0;i<n;i++) g[i]+=(c[i].rr+0.5);
//  for(int i=0;i<n;i++) cout<<(int)(c[i].rr+0.5)<<" ";cout<<endl;

for(int i=0;i<n;i++) g[i]=(g[i]+1)/2;
//  for(int i=0;i<n;i++) cout<<g[i]<<" ";cout<<endl;

Manacher(t);
//  for(int i=0;i<n;i++) cout<<f[i]<<" ";cout<<endl;

for(int i=0;i<l*2+1;i++) if(!(i&1)) f[i]-=1;
//  for(int i=0;i<n;i++) cout<<f[i]<<" ";cout<<endl;

for(int i=0;i<l*2-1;i++) ans=(ans+Pow(2,g[i])-(f[i+1]+1)/2-1)%p;

cout<<ans<<endl;

return 0;
}
/*
abaabaa
14

aaabbbaaa
44

aaaaaaaa
53
*/


posted @ 2017-01-05 11:06  北北北北屿  阅读(136)  评论(0编辑  收藏  举报