【TJOI 2019】甲苯先生的字符串

传送门

---

problem

给定字符串 $s_1$，你想求出满足下列限制的字符串 $s_2$ 的数量：

1. $s_2$ 的长度为 $n$。
2. $s_1$ 中的相邻的两个字母不能在 $s_2$ 中相邻地出现。

其中 $s_1$ 和 $s_2$ 仅包含小写字母 $a\sim z$。答案对 $10^9+7$ 取模。

数据范围：$n\le 10^{15}$。

---

solution

我们用 $(uv)$ 表示字符串 $“uv”$ 是否是 $s_1$ 的子串。若是，则 $(uv)=0$，否则 $(uv)=1$。

那么如果用 $f[i][v]$ 表示在第 $i$ 位填字符 $v$ 的方案数，那么有一个比较显然的转移：

$$f[i][v]=\sum _{u{=}^{\prime }{a}^{\prime }}^{{}^{\prime }{z}^{\prime }}[(uv)=1]f[i-1][u]$$

答案就是 ${\sum }_{v{=}^{\prime }{a}^{\prime }}^{{}^{\prime }{z}^{\prime }}f[n][v]$。

但是 $n$ 太大了，直接枚举转移显然是不可行的。这个时候注意到，每一步的转移方程都是一样的。

于是，我们就可以从矩阵的角度出发来优化这一转移。也即，把 $f[i-1]$ 看成行向量 $A$，$(uv)$ 构成的矩阵为转移矩阵 $B$，那么 $A\ast B$ 的得到的新的行向量就是 $f[i]$。这样就可以矩阵快速幂优化了。

时间复杂度 $O(26^3\log n)$。

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define P 1000000007
#define ll long long
using namespace std;
ll n;
char S[100005];
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
struct matrix{
	int m[26][26];
	matrix(int t=0){
		memset(m,0,sizeof(m));
		for(int i=0;i<26;++i)  m[i][i]=t;
	}
	friend matrix operator*(const matrix &A,const matrix &B){
		matrix C(0);
		for(int i=0;i<26;++i)
			for(int k=0;k<26;++k)
				for(int j=0;j<26;++j)
					C.m[i][j]=add(C.m[i][j],mul(A.m[i][k],B.m[k][j]));
		return C;
	}
	friend matrix operator^(matrix A,ll B){
		matrix ans(1);
		for(;B;B>>=1,A=A*A)  if(B&1)  ans=ans*A;
		return ans;
	}
}A,B;
int main(){
	scanf("%lld%s",&n,S+1);
	int len=strlen(S+1),ans=0;
	for(int i=0;i<26;++i)  A.m[0][i]=1;
	for(int i=0;i<26;++i)  for(int j=0;j<26;++j)  B.m[i][j]=1;
	for(int i=1;i<len;++i)  B.m[S[i]-'a'][S[i+1]-'a']=0;
	A=A*(B^(n-1));
	for(int i=0;i<26;++i)  ans=add(ans,A.m[0][i]);
	printf("%d\n",ans);
	return 0;
}