luogu P5337 [TJOI2019]甲苯先生的字符串

背景：

要中考改卷，搬到了国际部的机房。
鼠标滚轮是坏的，差评。
一直不能上$\text{luogu}$的原因是$\text{DNS}$域名有问题。

题目传送门：

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

题意：

给出$s1,n$，求长度为$n$的字符串$s2$使得其相邻字母不与$s1$中的相邻字母重复（注意顺序）。

思路：

设$f_{i,j}$表示当前做到第$i$位选字母$j$的方案数。
显然有方程：$f_{i,j}={\sum }_k{f}_{i-1,k}[k,j不相邻]$。
显然时间复杂度为$\Theta (26^{2}n)$，显然行不通。

考虑$dp$的优化。
一种显然的思想是矩阵乘法。
考虑上面的性质，不妨搞一个$26\ast 26$的矩阵来表示是否可以从$k$转移到$j$（其实你的$30pts$的做法也是如此）。
那就有：
$$\left[\begin{array}{c}base\end{array}\right]\ast \left[\begin{array}{c}{f}_{i,a}\\ f_{i,b}\\ f_{i,c}\\ ...\\ f_{,z}\end{array}\right]=\left[\begin{array}{c}{f}_{i+1,a}\\ f_{i+1,b}\\ f_{i+1,c}\\ ...\\ f_{i+1,z}\end{array}\right]$$
其中$base$就是是否可以转移的矩阵。
注意这里不像原来一样$f$在前，转移矩阵在后，因为$f_{i+1,a}$从$f_{i,a},f_{i,b},f_{i.c},...,f_{i,z}$转移到，符合矩阵乘法中的$(i,j)=(i,k)\ast (k,j)$的思想。
注意$f$的初始值是$1$。
在代码中$f=op$。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 1000000007
#define LL long long
using namespace std;
struct node
{
	LL a[30][30];
	void clear()
	{
		memset(a,0,sizeof(a));
	}
	friend node operator * (const node &a,const node &b)
	{
		node c;
		c.clear();
		for(int i=1;i<=26;i++)
			for(int j=1;j<=26;j++)
				for(int k=1;k<=26;k++)
					c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j]%mod)%mod;
		return c;
	}
} base,ans,op;
	LL n;
	int l;
	char s[100010];
	LL bz[30][30];
node work(LL x,node base)
{
	node ans;
	ans.clear();
	for(int i=1;i<=26;i++)
		ans.a[i][i]=1;
	while(x)
	{
		if(x&1) ans=ans*base;
		base=base*base;
		x>>=1;
	}
	return ans;
}
int main()
{
	scanf("%lld",&n);
	scanf("%s",s+1);
	l=strlen(s+1);
	op.clear();
	for(int i=1;i<=26;i++)
	{
		op.a[i][1]=1;
		for(int j=1;j<=26;j++)
			bz[i][j]=1;
	}
	for(int i=2;i<=l;i++)
		bz[s[i-1]-'a'+1][s[i]-'a'+1]=0;
	memcpy(base.a,bz,sizeof(base.a));
	ans=work(n-1,base)*op;
	LL sum=0;
	for(int i=1;i<=26;i++)
		sum=(sum+ans.a[i][1])%mod;
	printf("%lld\n",sum);
}