luogu P4199 万径人踪灭

背景:

FFT的题好神仙。

题目传送门：

https://www.luogu.org/problemnew/show/P4199

题意:

给一个字符串，求其回文子序列的个数（不能连续且位置也需要对称）。

思路：

懒人模式：https://www.cnblogs.com/MashiroSky/p/6341296.html

代码：

$longlong$的问题调了我$2h$（菜…）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
#define mod 1000000007
using namespace std;
const double pai=acos(-1.0);
struct comp
{
	double x,y;
	comp(double xx=0,double yy=0):x(xx),y(yy) {}
	friend comp operator+(const comp &x,const comp &y) {return comp(x.x+y.x,x.y+y.y);}
	friend comp operator-(const comp &x,const comp &y) {return comp(x.x-y.x,x.y-y.y);}
	friend comp operator*(const comp &a,const comp &b) {return comp(a.x*b.x-a.y*b.y,a.x*b.y+b.x*a.y);}
}a[500000],b[500000];
	int limit=1,n,l=0,ma=0;
	int r[500000];
	char st[500000],s[500000];
	int p[500000];
	LL pow2[500000],f[500000];
void init()
{
	while(limit<=ma)
		limit<<=1,l++;
	for(int i=1;i<=limit;i++)
		r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void FFT(comp *now,int ty)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		comp wn(cos(pai/mid),ty*sin(pai/mid));
		for(int j=0,R=(mid<<1);j<limit;j+=R)
		{
			comp w(1,0);
			for(int k=0;k<mid;k++,w=w*wn)
			{
				comp x=now[j+k],y=w*now[j+k+mid];
				now[j+k]=x+y;
				now[j+k+mid]=x-y;
			}
		}
	}
}
void work()
{
	for(int i=0;i<=limit;i++)
		a[i]=(comp){0,0};
	for(int i=0;i<n;i++)
		if(s[i]=='a') a[i].x=1; else a[i].x=0;
	FFT(a,1);
	for(int i=0;i<=limit;i++)
		b[i]=a[i]*a[i];

	for(int i=0;i<=limit;i++)
		a[i]=(comp){0,0};
	for(int i=0;i<n;i++)
		if(s[i]=='b') a[i].x=1; else a[i].x=0;

	FFT(a,1);
	for(int i=0;i<=limit;i++)
		b[i]=b[i]+(a[i]*a[i]);

	FFT(b,-1);
	for(int i=2;i<=ma;i++)
		f[i]+=(LL)floor(b[i-2].x+0.5)/limit;
}
LL manacher()
{
	int len=(n<<1)+2,id=0,ma=0;
	LL sum=0;
	st[0]='!';
	st[1]='#';
	for(int i=0;i<len;i++)
		st[i*2+2]=s[i],st[i*2+3]='#';
	for(int i=1;i<len;i++)
	{
		p[i]=ma>i?min(p[2*id-i],ma-i):1;
		for(;st[i+p[i]]==st[i-p[i]];p[i]++);
		if(p[i]+i>ma)
		{
			ma=p[i]+i;
			id=i;
		}
		sum=(sum+(p[i]>>1))%mod;
	}
	return sum;
}
int main()
{
	pow2[0]=1;
	for(int i=1;i<=400000;i++)
		pow2[i]=(pow2[i-1]<<1)%mod;
	scanf("%s",s);
	n=strlen(s);
	ma=(n<<1);
	init();
	work();
	LL tot=0;
	for(int i=2;i<=ma;i++)
		tot=((tot+pow2[(f[i]+1)>>1])%mod-1+mod)%mod;
	printf("%lld",(tot-manacher()+mod)%mod);
}