【TJOI2019】【矩阵快速幂】甲苯先生的字符串

【描述】
一天小甲苯得到了一条神的指示，他要把神的指示写下来，但是又不能泄露天机，所以他要用一种方法把神的指示记下来。神的指示是一个字符串，记为字符串s1，s1仅包含小写字母a−z。现在小甲苯想要写下神的指示，记为字符串s2，s2 仅包含小写字母a−z，要求s1中的相邻的两个字母不能在s2中相邻地出现。现在给定s2 的长度，小甲苯想知道他有多少种方法可以将神的指示写下来。输出种类数结果对109+7取模。

【输入】
第一行只有一个正整数n，代表字符串s2的长度，n≤1015
第二行是一个字符串，代表字符串s1 。s1 的长度不超过100000。

【输出】
输出一个整数，表示小甲苯可以写出的字符串的总数。结果对109+7取模

【样例输入】
2
ab
【样例输出】
675
水题配水样例

【思路】

这道题的思路还是很明显的。我们考虑构造这样一个模型：有26个点，有些点之间有边。问从某个点出发走t步到某个点的方案数。我们就可以定义一个矩阵$f[i][j]$表示从i出发走k步到j的方案数，其中，k是这个矩阵本身的属性。那么转移就可以利用乘法原理。设走2k步的方案数为$g[i][j]$，那么：
$$g[u][v]=\sum f[u][k]\ast f[k][v]$$
这就是一个乘法原理，同时恰好也是矩阵乘法，所以我们就可以优化转移。这道题恰好符合这个模型，相邻两个字符的联系就是一条边，s1的限制就是有些点之间没有边，剩下的点之间都有边。答案就是走n-1步的方案的矩阵的所有位置的和。

#include<bits/stdc++.h>
#define re register
using namespace std;
const int N=3e5+5;
const int mod=1e9+7;
long long n;
char s[N];
struct mat{
	int a[30][30];
	inline const int *const operator[](const int &offset)const{return a[offset];}
	inline int *const operator[](const int &offset){return a[offset];}
	friend inline mat operator*(const mat&a,const mat&b){
		mat c;memset(c.a,0,sizeof(c.a));
		for(int re i=0;i^26;++i)
			for(int re j=0;j^26;++j)
				for(int re k=0;k^26;++k)
					if((c[i][j]+=1ll*a[i][k]*b[k][j]%mod)>=mod)
						c[i][j]-=mod;
		return c;
	}
}a,b;
int main(){
	scanf("%lld",&n);scanf("%s",s);
	for(int re i=0;i^26;i++)
		for(int re j=0;j^26;j++)
			a[i][j]=1;
	for(int re i=0;i^26;++i)b[i][i]=1;
	int len=strlen(s)-1;
	for(int re i=0;i^len;i++)a[s[i]-'a'][s[i+1]-'a']=0;
	--n;
	while(n){if(n&1)b=b*a;a=a*a;n>>=1;}
	int ans=0;
	for(int re i=0;i^26;++i)
		for(int re j=0;j^26;++j)
			(ans+=b[i][j])%=mod;
	cout<<ans<<"\n";
}