万径人踪灭

VII.万径人踪灭

千山鸟飞绝,万径人踪灭。
孤舟蓑笠翁,独钓寒江雪。
  ——【唐】柳宗元《江雪》

跑题了跑题了

我们可以知道,

\(\text{答案=位置对称且字符对称的子序列的数量-回文子串数}\)

关于回文子串数,我们可以使用Manacher算法在\(O(n)\)时间内实现。如果不会的话,可以参加鄙人的拙作,这里不再赘述。

那么如何求出位置对称且字符对称的子序列的数量呢?

考虑如果对于某个对称中心来说,有\(k\)对字符关于它位置对称并且字符对称,这个对称中心的贡献是多少。

当这个对称中心是一个具体的字符时,贡献为\(2^{k+1}-1\),这\(k\)对字符,连带它自己,一共\(k+1\)个物品,都有选与不选两种情况,共\(2^{k+1}\)中方案。当然,不能一个物品都不选,因此要减去\(\varnothing\)时的方案,即减一。

当这个对称中心是两个字符中间的夹缝时,贡献为\(2^k-1\)。理由同上。

考虑将\(a\)\(b\)分开考虑。以考虑\(a\)为例。则我们令\(f(x)=[s_x='a']\),用\(f\)自乘。我们看看结果有何含义。

\(f(x)=f(y)=1\)时,\(f(x+y)\)会加一。而它的实际含义是,\(\dfrac{x+y}{2}\)处的\(k\)增加了\(1\)。也就是说,\(f(i)\)的实际含义是\(\dfrac{i}{2}\)处的\(k\)值。

考虑\(b\)时同理,我们令\(g(x)=[s_x='b']\),再用\(g\)自乘。之后,将\(f\)\(g\)加在一起,得到每个位置的\(k\)

但这个\(k\)并不是真实的\(k\)。在卷积时,\(f(x)f(y)\)\(f(y)f(x)\),除非\(x=y\),否则都被考虑了两次。因此,如果\(x\)是一个字符,\(h(x)\)必定为奇,因为除了它自己被考虑一次以外,其他的对都被考虑两次。如果\(x\)是一个夹缝,\(h(x)\)必定为偶,因为所有对都被考虑两次。

这样的话,我们可以令答案增加\(\large2^{\left\lfloor{\frac{h(x)+1}{2}}\right\rfloor}-1\),当\(h(x)\)为奇时,实际上增加了\(1\);当\(h(x)\)为偶数时,实际上什么也没有增加。

代码:

#include<bits/stdc++.h>
using namespace std;
const double pi=acos(-1);
const int mod=1e9+7;
int S,lim=1,lg,rev[1<<20],h[1<<20],pov[1<<20],res,rad[1<<20];
char s[1<<20];
struct cp{
	double x,y;
	cp(double u=0,double v=0){x=u,y=v;}
	friend cp operator +(const cp &u,const cp &v){return cp(u.x+v.x,u.y+v.y);}
	friend cp operator -(const cp &u,const cp &v){return cp(u.x-v.x,u.y-v.y);}
	friend cp operator *(const cp &u,const cp &v){return cp(u.x*v.x-u.y*v.y,u.x*v.y+u.y*v.x);}
}f[1<<20],g[1<<20];
void FFT(cp *a,int tp){
	for(int i=0;i<lim;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
	for(int md=1;md<lim;md<<=1){
		cp rt=cp(cos(pi/md),tp*sin(pi/md));
		for(int stp=md<<1,pos=0;pos<lim;pos+=stp){
			cp w=cp(1,0);
			for(int i=0;i<md;i++,w=w*rt){
				cp x=a[pos+i],y=w*a[pos+md+i];
				a[pos+i]=x+y;
				a[pos+md+i]=x-y;
			}
		}
	}
}
void Prepare(){
	for(int i=S-1;i>=0;i--)s[2*i+1]=s[i];
	S=2*S+1;
	for(int i=0;i<S;i+=2)s[i]='#';
	s[S]='\0';
}
void Manacher(){
	Prepare();
	int Rpos=-1,Centre=-1;
	for(int i=0;i<S;i++){
		rad[i]=(i<Rpos)?min(Rpos-i,rad[2*Centre-i]):1;
		while(i-rad[i]>=0&&i+rad[i]<S)if(s[i-rad[i]]==s[i+rad[i]])rad[i]++;else break;
		if(i+rad[i]>Rpos)Rpos=i+rad[i],Centre=i;
		res-=rad[i]>>1;
		if(res<0)res+=mod;
	}
}
int main(){
	scanf("%s",s),S=strlen(s);
	while(lim<=S*2)lim<<=1,lg++;
	for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
	pov[0]=1;
	for(int i=1;i<lim;i++)pov[i]=(pov[i-1]<<1)%mod;
	for(int i=0;i<S;i++)f[i]=cp(s[i]=='a',0),g[i]=cp(s[i]=='b',0);
	FFT(f,1),FFT(g,1);
	for(int i=0;i<lim;i++)f[i]=f[i]*f[i],g[i]=g[i]*g[i];
	FFT(f,-1),FFT(g,-1);
	for(int i=0;i<lim;i++)h[i]=(int)(f[i].x/lim+0.5)+(int)(g[i].x/lim+0.5),res=(res+pov[(h[i]+1)>>1]-1)%mod;
	Manacher(); 
	printf("%d\n",res);
	return 0;
}
posted @ 2020-04-25 11:49  Troverld  阅读(122)  评论(0编辑  收藏  举报