矩阵快速幂优化DP+例题详细分析

矩阵快速幂性质
快速求解线性递推式的结果，如斐波那契递推
优化的 DP 需要满足的条件
转移方程为线性递推式
转移次数超级多（n 很大）

例题1

衣食无忧的 Q老师 有一天突发奇想，想要去感受一下劳动人民的艰苦生活。
具体工作是这样的，有 N 块砖排成一排染色，每一块砖需要涂上红、蓝、绿、黄这 4 种颜色中的其中 1 种。且当这 N 块砖中红色和绿色的块数均为偶数时，染色效果最佳。
为了使工作效率更高，Q老师 想要知道一共有多少种方案可以使染色效果最佳，你能帮帮他吗？Input第一行为 T，代表数据组数。(1 ≤ T ≤ 100)
接下来 T 行每行包括一个数字 N，代表有 N 块砖。(1 ≤ N ≤ 1e9)Output输出满足条件的方案数，答案模 10007。
Sample Input
2
1
2
Sample Output
2
6
连续格子染色，很明显有子结构的性质，可以考虑 DP。我们可以需要考虑DP的状态转移方程：
题目问 n 个格子，红绿均为偶数的染色方案数
因此令 A[i] 表示 i 个格子，红绿均为偶数的染色方案数
只有 A[i] 是否足够？显然是不够的，因为俩偶数也可以从一奇一偶转移过来。
因此我们进行 DP 状态添加：
B[i] 表示 i 个格子，红绿均为奇数的染色方案数
C[i] 表示 i 个格子，红绿有一个为偶数的染色方案数
DP 转移方程：
根据 DP 状态，仔细思考转移情况即可得到
A[i] = 2 * A[i-1] + C[i-1]
B[i] = 2 * B[i-1] + C[i-1]
C[i] = 2 * A[i-1] + 2 * B[i-1] + 2 * C[i-1]
由于n是1e9所以O（n)的dp是不行的，所以需要进行优化——矩阵快速幂优化。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 10007
using namespace std;
int T,N;
struct matrix{
 int x[5][5];
 matrix operator*(const matrix &t)const{
  matrix ret;
  for(int i=0;i<3;i++){
   for(int j=0;j<3;j++){
    ret.x[i][j]=0;
    for(int k=0;k<3;k++){
     ret.x[i][j]+=x[i][k]*t.x[k][j]%mod;
     ret.x[i][j]%=mod;
    }
   }
  }
  return ret;
 }
 matrix(){ memset(x,0,sizeof(x));}
 matrix(const matrix& t){memcpy(x,t.x,sizeof(x));}
};
matrix solve(matrix a,int x)
{
 matrix ret;
 for(int i=0;i<3;i++)ret.x[i][i]=1;
 while(x){
  if(x&1)ret=ret*a;
  a=a*a;
  x>>=1;
 }
 return ret;
}
int ans[5];
int main(){
 cin>>T;
 ans[0]=2,ans[1]=0,ans[2]=2;
 while(T--){
  cin>>N;
  matrix u;
  u.x[0][0]=2,u.x[0][1]=0,u.x[0][2]=1;
  u.x[1][0]=0,u.x[1][1]=2,u.x[1][2]=1;
  u.x[2][0]=2,u.x[2][1]=2,u.x[2][2]=2;
  matrix tmp=solve(u,N-1);
  int sum=0;
  for(int i=0;i<3;i++)sum+=tmp.x[0][i]*ans[i]%mod;
  sum%=mod;
  cout<<sum<<endl;
 }
 return 0;
}

例题二

忙碌了一个学期的 Q老师 决定奖励自己 N 天假期。
假期中不同的穿衣方式会有不同的快乐值。
已知 Q老师 一共有 M 件衬衫，且如果昨天穿的是衬衫 A，今天穿的是衬衫 B，则 Q老师 今天可以获得 f[A][B] 快乐值。
在 N 天假期结束后，Q老师 最多可以获得多少快乐值？Input输入文件包含多组测试样例，每组测试样例格式描述如下：
第一行给出两个整数 N M，分别代表假期长度与 Q老师 的衬衫总数。（2 ≤ N ≤ 100000, 1 ≤ M ≤ 100）
接下来 M 行，每行给出 M 个整数，其中第 i 行的第 j 个整数，表示 f[i][j]。（1 ≤ f[i][j] ≤ 1000000）
测试样例组数不会超过 10。Output每组测试样例输出一行，表示 Q老师 可以获得的最大快乐值。Sample Input3 2
0 1
1 0
4 3
1 2 3
1 2 3
1 2 3
Sample Output2
9
天数连续，很明显有子结构的性质，可以考虑 DP
n 不是很大，因此不一定是矩阵快速幂优化，先列出 DP 状态和 DP 转移方程再进行判断

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m;
int f[110][110];
struct matrix{
 long long x[110][110];
 matrix operator*(const matrix &t)const{
  matrix ret;
  for(int i=1;i<m;i++){
   for(int j=0;j<m;j++){
    ret.x[i][j]=0;
    for(int k=0;k<m;k++){
     ret.x[i][j]=max(ret.x[i][j],x[i][k]+t.x[k][j]);
    }
   }
  }
  return ret;
 }
 matrix(){ memset(x,0,sizeof(x));}
 matrix(const matrix& t){memcpy(x,t.x,sizeof(x));}
};
matrix solve(matrix a,int x)
{
 matrix ret=a;
 while(x){
  if(x&1)ret=ret*a;
  a=a*a;
  x>>=1;
 }
 return ret;
}
long long ans=0;
int main(){
 while(cin>>n>>m){
  matrix u;
  for(int i=0;i<m;i++){
   for(int j=0;j<m;j++){
    cin>>f[i][j];
    u.x[i][j]=f[i][j];
   }
  }
  matrix tmp=solve(u,n-2);
  ans=0;
  for(int i=0;i<m;i++){
   for(int j=0;j<m;j++){
    ans=max(ans,tmp.x[i][j]);
   }
  }
  cout<<ans<<endl;
 }
 return 0;
}