因为没有简化题意一直没去做,直到今天讲这道题才口胡出来
要求对称,很明显这样一个“子序列”的对称中心只可能有一个,那么先枚举这个对称中心。
然后我们需要判断两个位置是否相同。看上去好像很困难。
考虑设计哈希函数 \(f(x,y)\),使得 \(f(V_a,V_b)\neq 0\) 且 \(f(V_a,V_a)=f(V_b,V_b)=0\)。
首先等于 \(0\),那么一定有一个 \((x-y)\)。其次,注意到字符集只有 \(2\),为了方便,我们令 \(f(x,y)=-f(y,x)\)。那么就包含了 \((x-y)^2\)。
所以 \(f(x,y)=(V_x-V_y)^2=V_x^2+V_y^2-2V_xV_y\),这就够了。
我们需要计算的是 \(match[i]=\sum_{j=0}f(i-j,i+j)\),这有点像卷积。
拆开得到:
\[\sum_{j=0}(V_{i-j}^2+V_{i+j}^2)-2V_{i+j}V_{i-j} \]后面这玩意儿可以用前缀和搞定,前面这玩意儿做一个卷积就好啦。
这样做会算上连续的一段,使用 manacher 计算出来并且减掉就好啦。
复杂度 \(O(n\log n)\)。
#include<cstring>
#include<cstdio>
typedef unsigned ui;
const ui M=1e5+5,G=3,mod=998244353,MOD=1e9+7;
ui n,V[M],S[M],pw2[M],f[M<<2];ui ans;char s[M];
ui p[M<<1];char c[M<<1];
inline ui min(const ui&a,const ui&b){
return a>b?b:a;
}
inline ui pow(ui a,ui b=mod-2){
ui ans(1);for(;b;b>>=1,a=1ull*a*a%mod)if(b&1)ans=1ull*ans*a%mod;return ans;
}
inline ui Add(const ui&a,const ui&b){
return a+b>=mod?a+b-mod:a+b;
}
inline ui Del(const ui&a,const ui&b){
return b>a?a-b+mod:a-b;
}
inline void swap(ui&a,ui&b){
ui c=a;a=b;b=c;
}
inline void DFT(ui*f,const ui&n){
for(ui len=n>>1;len>=1;len>>=1){
const ui w1=pow(G,(mod-1>>1)/len);
for(ui k=0;k<n;k+=len<<1){
for(ui w(1),i=0;i<len;++i){
const ui x=f[i|k],y=f[i|k|len];
f[i|k]=Add(x,y);f[i|k|len]=1ull*w*Del(x,y)%mod;
w=1ull*w*w1%mod;
}
}
}
}
inline void IDFT(ui*f,const ui&n){
for(ui len=1;len<n;len<<=1){
const ui w1=pow(G,(mod-1>>1)/len);
for(ui k=0;k<n;k+=len<<1){
for(ui w(1),i=0;i<len;++i){
const ui x=f[i|k],y=1ull*w*f[i|k|len]%mod;
f[i|k]=Add(x,y);f[i|k|len]=Del(x,y);
w=1ull*w*w1%mod;
}
}
}
const ui inv=pow(n);
for(ui i=0;i<n;++i)f[i]=1ull*f[i]*inv%mod;
for(ui i=1;(i<<1)<n;++i)swap(f[i],f[n-i]);
}
inline ui manacher(char*s){
ui R(0),mid(0),ans(0);c[0]='^';c[1]='#';
for(ui i=1;i<=n;++i)c[i<<1]=s[i],c[i<<1|1]='#';
for(ui i=1;i<=(n<<1);++i){
p[i]=i<=R?min(p[(mid<<1)-i],p[mid]+mid-i):1;
while(c[i-p[i]]==c[i+p[i]])++p[i];
if(i+p[i]-1>R)mid=i,R=i+p[i]-1;
ans=(ans+(p[i]>>1))%MOD;
}
return ans;
}
signed main(){
scanf("%s",s+1);n=strlen(s+1);pw2[0]=1;
for(ui i=1;i<=n;++i){
V[i]=s[i]=='a'?1:2;pw2[i]=pw2[i-1]*2%MOD;
f[i]=V[i];S[i]=S[i-1]+V[i]*V[i];
}
ui len(1);
while(len<n+n+1)len<<=1;
DFT(f,len);
for(ui i=0;i<len;++i)f[i]=1ull*f[i]*f[i]%mod;
IDFT(f,len);
for(ui i=1;i<=n;++i){
ui len=min(n-i,i-1);
ans=(ans+pw2[len+1-((S[i+len]-S[i-len-1])-f[i<<1])]-1)%MOD;
if(len!=n-i)++len;
ans=(ans+pw2[len-((S[i+len]-S[i-len])-f[i<<1|1])]-1)%MOD;
}
printf("%u",(MOD+ans-manacher(s))%MOD);
}