[CF506E]Mr. Kitayuta's Gift

Description

给出一个字符串s，你需要往其中插入n个小写字符得到字符串t，使得t是一个回文串
问所有能得到的本质不同的t的个数
|s|<=200,n<=10^9

Solution

先考虑最暴力的做法，我们设F[l][r][k]表示当前s[l…r]还没有被匹配，从外往里做到第k层
转移直接考虑第k+1层选什么字符，视情况可以转移到F[l+1][r-1][k+1],F[l+1][r][k+1],F[l][r-1][k+1],F[l][r][k+1]
可以发现我们实际上是在一个FSM上走路，问走(n+|s|)/2步到终止点的方案数

容易发现这个自动机只有两种节点：红点(出度为2)和绿点(出度为1)
考虑我们走路的实质是，先选择一条链，再在这条链上走路，并且不同的链的方案数只和其上的红点/绿点数量有关
对于一个有i个红点的链，容易得知有$\lceil \frac{|s|-i}{2}\rceil$个绿点，也就是说根据红点的数量本质不同的链只有O(|s|)种
但是对于每种链矩乘是$O(|s{|}^4\log n)$的还是过不了
其实我们可以对所有链一起矩乘
建图的话建两排点，第一排红点第二排绿点，每个红点向第$\lceil \frac{|s|-i}{2}\rceil$个绿点和下一个红点连边，这样就相当于所有情况一起考虑了
注意当t串长为奇数的时候要特殊处理
复杂度$O(|s{|}^3\log n)$

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

const int N=205,Mo=10007;

void inc(int &x,int y) {x=x+y>=Mo?x+y-Mo:x+y;}

int n,m,t,f[N][N][N],g[N];
char st[N];

struct Matrix{
	int a[305][305];
	friend Matrix operator * (Matrix a,Matrix b) {
		Matrix c;memset(c.a,0,sizeof(c.a));
		fo(k,1,t) fo(i,1,t) fo(j,1,t) c.a[i][j]+=a.a[i][k]*b.a[k][j]%Mo;
		fo(i,1,t) fo(j,1,t) c.a[i][j]%=Mo;
		return c;
	}
}trs;

Matrix pwr(Matrix a,int y) {
	Matrix b;memset(b.a,0,sizeof(b.a));
	fo(i,1,t) b.a[i][i]=1;
	for(;y;y>>=1,a=a*a) if (y&1) b=b*a;
	return b;
}

int main() {
	scanf("%s",st+1);n=strlen(st+1);scanf("%d",&m);
	f[1][n][0]=1;
	fo(i,1,n)
		fd(j,n,i) 
			if (st[i]==st[j]) {
				fo(k,0,i+n-j-1) {
					if (i+1<=j-1) inc(f[i+1][j-1][k],f[i][j][k]);
					else inc(g[k],f[i][j][k]);
				}
			} else {
				fo(k,0,i+n-j-1) {
					inc(f[i+1][j][k+1],f[i][j][k]);
					inc(f[i][j-1][k+1],f[i][j][k]);
				}
			}
	t=n+(n+1)/2+1;
	trs.a[t][t]=26;
	fo(i,1,n) {
		trs.a[i][i]=24;trs.a[i][t-(n-i+1)/2]=g[i];
		if (i<n) trs.a[i][i+1]=1;
	}
	fo(i,n+1,t-1) trs.a[i][i+1]=1,trs.a[i][i]=25;
	trs=pwr(trs,(n+m+1)>>1);
	int ans=(trs.a[1][t]+trs.a[n+1][t]*g[0])%Mo;
	if ((n+m)&1) {
		fo(i,1,t) fo(j,1,t) trs.a[i][j]=0;
		fo(i,0,n) g[i]=0;
		fo(i,1,n) if (st[i]==st[i+1]) fo(k,0,n) inc(g[k],f[i][i+1][k]);
		fo(i,1,n) {
			trs.a[i][i]=24;trs.a[i][t-(n-i+1)/2]=g[i];
			if (i<n) trs.a[i][i+1]=1;
		}
		fo(i,n+1,t-1) trs.a[i][i+1]=1,trs.a[i][i]=25;
		trs=pwr(trs,(n+m+1)>>1);
		ans=(ans-trs.a[1][t]-trs.a[n+1][t]*g[0])%Mo;
	}
	printf("%d\n",(ans+Mo)%Mo);
	return 0;
}