bzoj3160 万径人踪灭(manacher+fft)

要求不连续的回文子序列个数。

我们考虑用所有的回文子序列个数-连续的回文子序列个数。

连续的回文子序列个数就是回文子串的个数,我们可以用manacher来求。

考虑如何求回文子序列的总数。我们先都按manacher那样插入若干’#’以后,处理出一个f[i]数组,表示i两侧对称的字母对数。(含自己,不含#)。则总数就是 i=22n2f[i]1

如何处理f数组呢?我们发现若一对位置的字母对f[i]有贡献,则这对位置在原串中的标号和为i。我们可以画图发现这一事实。于是有
f[i]=j=1i1[s[j]==s[ij]]+1>>1

这个东西就是个卷积,不过怎么算呢?我们可以先算a的贡献:把a的位置置为1,b的位置置为0.则 j=0ia[j]a[ij] 就是a的贡献。然后再算b的贡献:把b的位置置为1,a的位置置为0.然后就可以得到答案了。

我一开始太naive了,真的分开算了6遍fft。。。其实你可以在点值表达直接做加法,再最后一次IDFT回去。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <complex>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define pi acos(-1)
#define mod 1000000007
typedef complex<double> E;
int n,m,f[N<<1],ans=0,L=0,R[N<<2],p[N<<1];
char s[N],s2[N<<1];
E a[N<<2],b[N<<2];
inline int ksm(int x,int k){
    int res=1;
    for(;k;k>>=1,x=(ll)x*x%mod)
        if(k&1) res=(ll)res*x%mod;return res;
}
inline void fft(E *a,int f){
    for(int i=0;i<n;++i) if(i<R[i]) swap(a[i],a[R[i]]);
    for(int i=1;i<n;i<<=1){
        E wn(cos(pi/i),f*sin(pi/i));
        for(int j=0,p=i<<1;j<n;j+=p){
            E w(1,0);
            for(int k=0;k<i;++k,w*=wn){
                E x=a[j+k],y=a[j+k+i]*w;
                a[j+k]=x+y;a[j+k+i]=x-y;
            }
        }
    }if(f==-1) for(int i=0;i<n;++i) a[i]/=n;
}
int main(){
//  freopen("1.in","r",stdin);
    scanf("%s",s+1);n=strlen(s+1);
    m=n+n;for(n=1;n<=m;n<<=1) ++L;
    for(int i=0;i<n;++i) R[i]=(R[i>>1]>>1)|(i&1)<<L-1;
    for(int i=1;i<=m/2;++i) a[i]=(s[i]=='a');fft(a,1);
    for(int i=0;i<n;++i) b[i]=a[i]*a[i];memset(a,0,sizeof(a));
    for(int i=1;i<=m/2;++i) a[i]=(s[i]=='b');fft(a,1);
    for(int i=0;i<n;++i) b[i]+=a[i]*a[i];fft(b,-1);
    for(int i=2;i<=m;++i) f[i]=(int)(b[i].real()+0.1)+1>>1;
    for(int i=2;i<=m;++i) ans=(ans+ksm(2,f[i])-1)%mod;
    n=0;m>>=1;
    for(int i=1;i<=m;++i) s2[++n]='#',s2[++n]=s[i];s2[++n]='#';
    int id=0,tot=0;
    for(int i=1;i<=n;++i){
        if(i<id+p[id]) p[i]=min(p[id*2-i],id+p[id]-i);
        else p[i]=1;
        while(i-p[i]>=1&&i+p[i]<=n&&s2[i-p[i]]==s2[i+p[i]]) ++p[i];
        if(i+p[i]>id+p[id]) id=i;
        tot=(tot+p[i]/2)%mod;
    }ans-=tot;if(ans<0) ans+=mod;
    printf("%d\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值