BZOJ3160: 万径人踪灭

一开始看错题以为位置可以不对称……

在串S相邻字符间插入#就可以处理偶数回文串了
不连续的回文子序列个数=回文子序列个数-连续的回文子序列个数
连续的可以用Manacher算
对于不连续的，若i是回文中心，i两侧对称的字符个数为c[i]，就有$2^{c[i]}-1$个

因为位置是对称，计算c[i]可以用fft，先令a的值为1卷一次再令b的值为1卷一次，注意回文中心s[i]是a或b的情况

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const int maxn = 580000;
const int mod = 1e9+7;
const double pi = acos(-1);

int n,N,ln;
struct E
{
    double x,y;
    friend inline E operator +(const E a,const E b){return (E){a.x+b.x,a.y+b.y};}
    friend inline E operator -(const E a,const E b){return (E){a.x-b.x,a.y-b.y};}
    friend inline E operator *(const E a,const E b){return (E){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};}
}s[maxn],zero;
int id[maxn];
void pre()
{
    N=1,ln=0; while(N<=(n<<1)) N<<=1,ln++;
    for(int i=1;i<N;i++) id[i]=id[i>>1]>>1|(i&1)<<(ln-1);
}
void DFT(E f[],int sig)
{
    for(int i=1;i<N;i++) if(i<id[i]) swap(f[i],f[id[i]]);
    for(int m=2;m<=N;m<<=1)
    {
        int t=m>>1;
        E wn=(E){cos(2*pi/m),sin(2*pi*sig/m)};
        for(int j=0;j<N;j+=m)
        {
            E w0=(E){1,0};
            for(int i=0;i<t;i++,w0=w0*wn)
            {
                E tx=f[j+i],ty=f[j+i+t]*w0;
                f[j+i]=tx+ty;
                f[j+i+t]=tx-ty;
            }
        }
    }
    if(sig==-1) for(int i=0;i<N;i++) f[i].x/=(double)N;
}

int m,len;
char str[maxn],S[maxn];
int rad[maxn];
int F[maxn],pw2[maxn],ans;

int main()
{
    //freopen("tmp.in","r",stdin);
    //freopen("tmp.out","w",stdout);

    scanf("%s",str); m=strlen(str);
    S[len=1]='!';
    for(int i=0;i<m;i++) S[++len]='#',S[++len]=str[i];
    S[++len]='#',S[++len]='?';

    pw2[0]=1; for(int i=1;i<=len;i++) pw2[i]=pw2[i-1]*2%mod;

    n=len; pre();
    for(int i=1;i<=len;i++) 
        s[i].x=S[i]=='a'?1:0;
    DFT(s,1);
    for(int i=0;i<N;i++) s[i]=s[i]*s[i];
    DFT(s,-1);
    for(int i=1;i<=len;i++) 
    {
        int temp=(int)(s[i<<1].x+0.5);
        if(S[i]=='a') temp++;
        F[i]+=temp>>1;
    }

    for(int i=0;i<N;i++) s[i]=zero;
    for(int i=1;i<=len;i++) s[i].x=S[i]=='b'?1:0;
    DFT(s,1);
    for(int i=0;i<N;i++) s[i]=s[i]*s[i];
    DFT(s,-1);
    for(int i=1;i<=len;i++) 
    {
        int temp=(int)(s[i<<1].x+0.5);
        if(S[i]=='b') temp++;
        F[i]+=temp>>1;
    }

    for(int i=1;i<=len;i++) (ans+=pw2[F[i]]-1)%=mod;

    //for(int i=1;i<=len;i++) printf("%c",S[i]); puts("");
    //for(int i=1;i<=len;i++) printf("%d",F[i]); puts("");

    int mx=0,ii=0;
    for(int i=2;i<=len;i++) 
    {
        if(mx>i) rad[i]=min(mx-i,rad[2*ii-i]);
        while(S[i-rad[i]]==S[i+rad[i]]) rad[i]++;
        if(i+rad[i]>mx) mx=i+rad[i],ii=i;

        (ans-=rad[i]>>1)%=mod;
    }


    //puts("");
    //for(int i=1;i<=len;i++) printf("%c",S[i]); puts("");
    //for(int i=1;i<=len;i++) printf("%d",rad[i]); puts("");
    (ans+=mod)%=mod;
    printf("%d\n",ans);

    return 0;
}