矩阵快速幂总结

矩阵快速幂

杭电培训总结,侵删

引导问题

一个有0和1组成的字符串，人和子串都不能包含$101$和$111$，求满足要求的长度为$L(1<=L<=10^8)$的字符串一共有多少，结果对$10^9+7$求模。

思路：

设$f(i)$为长度为$i$的合法字符串的数量。

可知$f(1)=2$ “1”，“0”

$f(2)=4$ “00”，“01”，“10”，“11”

$f(3)=6$ “000”，“001”，“010”，“011”，“100”，“101”，“110”，“111”

那么联想到$f(n)$，如果第$n$位取$0$，那么第$n-1$位和$n-2$位无论取什么，后三位也都不会构成$101$或者$111$。当第$n$位确定为$0$之后，$f(n)=f(n-1)$。

第$n$位取$1$，第$n-1$位和$n-2$位的组合有四种情况：00，01，10，11。10和11不能取。10和11和最后的一位1会组成非法字符串101和111。所以又分了两种子情况，第$n-1$位和$n-2$位为00和第$n-1$位和$n-2$位01的。由于00和第$n-3$位不会构成非法字符串,数量是$f(n-3)$。但是01情况下，如果第$n-3$位为1的，会组成101的非法串，所以第$n-3$位只能为0，由于第$n-2$位和$n-3$位的00不会构成非法串，该种情况数量为$f(n-4)$

递推公式为:
$$f(n)=f(n-1)+f(n-2)+f(n-4)$$

矩阵乘法

//两个n行n列的矩阵相乘
const int maxn = 50;
int mat1[maxn][maxn], mat2[maxn][maxn];
int res[maxn][maxn];
for(int i=1;i<=n;++i)
    for(int j=1;j<=n;++j)
        for(int k=1;k<=n;++k)
            res[i][j] += mat1[i][k]*mat2[k][j];
	

斐波那契数列

$$f(0)=0，f(n)=f(n-1)+f(n-2)$$

构造一个矩阵：

$$\left[\begin{array}{c}f[n]\\ f[n-1]\end{array}\right]=\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\end{array}\right]$$

设：

$$\begin{gathered} 设C(n)=\left[\begin{array}{c}f[n]\\ f[n-1]\end{array}\right] \\ B=\left[\begin{array}{cc}1& 1\\ 1& 0\end{array}\right] \end{gathered}$$

等价于
$$\begin{gathered} C(n)=B\ast C(n-1) \\ =B\ast B\ast C(n-2) \\ =B^{n-1}\ast C(1) \end{gathered}$$

①任意一个形如$f(n)=a\ast f(n-1)+b\ast f(n-2)$

$$构造矩阵：\left[\begin{array}{c}f[n]\\ f[n-1]\end{array}\right]=\left[\begin{array}{cc}a& b\\ 1& 0\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\end{array}\right]$$

例如$f(n)=3\ast f(n-1)+5\ast f(n-3)+9\ast f(n-4)$

构造：
$$\left[\begin{array}{c}f[n]\\ f[n-1]\\ f[n-2]\\ f[n-3]\end{array}\right]=\left[\begin{array}{cccc}3& 0& 5& 9\\ 1& 0& 0& 0\\ 0& 1& 0& 0\\ 0& 0& 1& 0\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\\ f[n-3]\\ f[n-4]\end{array}\right]$$

②$f(n)=a\ast f(n-1)+b\ast f(n-3)+c$

$$\left[\begin{array}{c}f[n]\\ f[n-1]\\ f[n-2]\\ 1\end{array}\right]=\left[\begin{array}{cccc}a& 0& b& c\\ 1& 0& 0& 0\\ 0& 1& 0& 0\\ 0& 0& 0& 1\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\\ f[n-3]\\ 1\end{array}\right]$$

或者
$$\left[\begin{array}{c}f[n]\\ f[n-1]\\ f[n-2]\\ c\end{array}\right]=\left[\begin{array}{cccc}a& 0& b& 1\\ 1& 0& 0& 0\\ 0& 1& 0& 0\\ 0& 0& 0& 1\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\\ f[n-3]\\ c\end{array}\right]$$

③$f(1)=1,f(2)=2,f(n)=f(n-1)+2f(n-2)+n^3$

根据二项式定理将$n^3$拆分为$n^3=(n-1+1{)}^3=(n-1{)}^3+3(n-1{)}^2+3(n-1)+1$

那么$n^2=(n-1{)}^2+2(n-1)+1$

$$\begin{gathered} f(n) \\ =f(n-1)+2f(n-2)+n^3 \\ =f(n-1)+2f(n-2)+(n-1{)}^3+3(n-1{)}^2+3(n-1)+1 \end{gathered}$$
$$\left[\begin{array}{c}f[n]\\ f[n-1]\\ n^3\\ n^2\\ n\\ 1\end{array}\right]=\left[\begin{array}{cccccc}1& 2& 1& 3& 3& 1\\ 1& 0& 0& 0& 0& 0\\ 0& 0& 1& 3& 3& 1\\ 0& 0& 0& 1& 2& 1\\ 0& 0& 0& 0& 1& 1\\ 0& 0& 0& 0& 0& 1\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\\ (n-1{)}^3\\ (n-1{)}^2\\ (n-1)\\ 1\end{array}\right]$$
例如:$f(n)=f(n-1)+2f(n-2)+n^3+n^2$
$$\left[\begin{array}{c}f[n]\\ f[n-1]\\ n^3\\ n^2\\ n\\ 1\end{array}\right]=\left[\begin{array}{cccccc}1& 2& 1& 4& 5& 2\\ 1& 0& 0& 0& 0& 0\\ 0& 0& 1& 3& 3& 1\\ 0& 0& 0& 1& 2& 1\\ 0& 0& 0& 0& 1& 1\\ 0& 0& 0& 0& 0& 1\end{array}\right]\ast \left[\begin{array}{c}f[n-1]\\ f[n-2]\\ (n-1{)}^3\\ (n-1{)}^2\\ (n-1)\\ 1\end{array}\right]$$

二项式定理:
$$\begin{gathered} (x+y{)}^n=\left(\begin{array}{c}n\\ 0\end{array}\right)x^n{y}^0+\left(\begin{array}{c}n\\ 1\end{array}\right)x^{n-1}{y}^1+\left(\begin{array}{c}n\\ 2\end{array}\right)x^{n-2}{y}^2+...+\left(\begin{array}{c}n\\ n-1\end{array}\right)x^1{y}^{n-1}+\left(\begin{array}{c}n\\ n\end{array}\right)x^0{y}^n \\ (x+y{)}^n=\sum _{k=0}^n\left(\begin{array}{c}n\\ k\end{array}\right)x^{n-k}{y}^k=\sum _{k=0}^n\left(\begin{array}{c}n\\ k\end{array}\right)x^k{y}^{n-k} \end{gathered}$$

---

---

④包含前缀和

$T[0]=T[1]=t[2]=1$

$T[n]=T[n-1]+T[n-2]+T[n-3](n>=3)$

给定$a$和$b$ $(1<=a<=b<=10^9)$,求$(T[a]+T[a+1]+...+T[b])\%(10^9+7)$

思路:$$\begin{gathered} 前缀和 \\ S[n]=S[n-1]+T[n] \\ S[n]=S[n-1]+T[n-1]+T[n-2]+T[n-3] \end{gathered}$$

构造矩阵:
$$\left[\begin{array}{c}S[n]\\ T[n]\\ T[n-1]\\ T[n-2]\end{array}\right]=\left[\begin{array}{cccc}1& 1& 1& 1\\ 0& 1& 1& 1\\ 0& 1& 0& 0\\ 0& 0& 1& 0\end{array}\right]\left[\begin{array}{c}S[n-1]\\ T[n-1]\\ T[n-2]\\ T[n-3]\end{array}\right]$$

---

---

⑤前缀平方和

$A[0]=1$,$A[1]=1$,$A[n]=X\ast A[n-1]+Y\ast A[n-2](n>=2)$

求$S[n]={\sum }_0^{n}A[i{]}^2$

分析:

$S[n]=s[n-1]+A[n{]}^2$

$A[n{]}^2=X^2\ast A[n-1{]}^2+Y^2\ast A[n-2{]}^2+2XY\ast A[n-1]A[n-2]$

$A[n]乘以A[n-1]:A[n]=X\ast A[n-1]+Y\ast A[n-2]=>A[n]A[n-1]=X\ast A[n-1{]}^2+Y\ast A[n-2]A[n-1]$
$$\left[\begin{array}{c}S[n]\\ A[n{]}^2\\ A[n-1{]}^2\\ A[n]A[n-2]\end{array}\right]=\left[\begin{array}{cccc}1& X^2& Y^2& 2XY\\ 0& X^2& Y^2& 2XY\\ 0& 1& 0& 0\\ 0& X& 0& Y\end{array}\right]\left[\begin{array}{c}S[n-1]\\ A[n-1{]}^2\\ A[n-2{]}^2\\ A[n-1]A[n-2]\end{array}\right]$$

例题

一个$01$循环串,长度为$L(L<=100)$，这个串每秒都会进行一次变换，变换规则是:如果左边是1，则改变自己的状态，否则保持不变。
给定初始状态，问 $n$秒以后这个串的状态。

分析:

定义状态$f(n,L)$表示n秒之后,第L个字符是0还是1;
$$f(n,L)=\{\begin{array}{l}f(n-1,L)(f(n-1,L-1)=0)\\ 1-f(n-1,L)(f(n-1,L-1)=1)\end{array}$$
简化:

如果$f(n-1,L-1)=0$; $f(n-1,L)=f(n-1,L)+0=f(n-1,L)+f(n-1,L-1)$

如果$f(n-1,L-1)=1$; $1-f(n-1,L)=f(n-1,L-1)-f(n-1,L)$

由于取值只有0和1,那么1+1=0

化简为$f(n,L)=f(n-1,L-1)+f(n-1,L)$

构造:
$$\left[\begin{array}{c}f(n,1)\\ f(n,2)\\ ...\\ f(n,L)\end{array}\right]=\left[\begin{array}{ccccc}1& 0& ...& 0& 1\\ 1& 1& ...& 0& 0\\ ...& ...& ...& 1& 0\\ 0& 0& ...& 1& 1\end{array}\right]\ast \left[\begin{array}{c}f(n-1,1)\\ f(n-1,2)\\ ...\\ f(n-1,L)\end{array}\right]$$

矩阵快速幂模板

const int maxn = 12; //最大矩阵规模
int mod = 9973; //模数
int n; //矩阵规模
//矩阵结构体
struct mat {
	int m[maxn][maxn];
	mat() //构造函数 单位矩阵 
	{ 
		memset(m,0,sizeof m);
		for (int i=1;i<=n;++i)
			m[i][i] = 1;
	}
    //[]运算符重载
	int* operator[] (int index)
	{
		return m[index];
	}
    //矩阵乘法
	mat operator*(mat& a)
	{
		mat res;
		for(int i=1;i<=n;++i)
		{
			for(int j=1;j<=n;++j)
			{
				res[i][j]=0;
				for(int k=1;k<=n;++k)
					res[i][j] = (res[i][j] + m[i][k] * a[k][j] % mod) % mod;
			}
		}
		return res;
	}
};

//快速幂
mat fastpow(mat a,int b)
{
	mat res;
	while(b) 
	{
		if(b&1) res = res * a;
		a = a * a;
		b >>= 1;
	}
	return res;
}
int main()
{
    /*Code*/
    return 0;
}