SDU 程序设计思维与实践 Week14 Q老师度假【广义矩阵快速幂优化DP】

Q老师度假

忙碌了一个学期的 Q老师 决定奖励自己 N 天假期。
假期中不同的穿衣方式会有不同的快乐值。
已知 Q老师 一共有 M 件衬衫，且如果昨天穿的是衬衫 A，今天穿的是衬衫 B，则 Q老师 今天可以获得 f[A][B] 快乐值。
在 N 天假期结束后，Q老师 最多可以获得多少快乐值？

输入

输入文件包含多组测试样例，每组测试样例格式描述如下：
第一行给出两个整数 N M，分别代表假期长度与 Q老师 的衬衫总数。（2 ≤ N ≤ 100000, 1 ≤ M ≤ 100）
接下来 M 行，每行给出 M 个整数，其中第 i 行的第 j 个整数，表示 f[i][j]。（1 ≤ f[i][j] ≤ 1000000）
测试样例组数不会超过 10。

输出

每组测试样例输出一行，表示 Q老师 可以获得的最大快乐值。

输入样例

3 2
0 1
1 0
4 3
1 2 3
1 2 3
1 2 3

输出样例

2
9

思路分析

• 状态定义：$f[i][j]$表示第i天传第j件衣服所获得的快乐值的最大总和。
• 状态转移方程:$f[i][j]=max(f[i-1][k]+H[k][j]),1\le k\le M$。
• 类比矩阵乘法定义，$c[i][j]=\sum (A[i][k]\ast B[k][j])$，发现将*改成了+，将$\sum$改成了$max$。
• 矩阵快速幂利用了运算满足结合律这一性质，只要某种运算满足结合律，都可以使用快速幂。
• 证明满足结合律：设A、B、C 分别为 $a\ast b、b\ast c、c\ast d$ 的矩阵
• 矩阵快速幂转换方法：
• 需要N-1次矩阵变换即可得到答案，矩阵快速幂初始为全0阵，不初始化为单位阵。
• 数据范围需要使用 $longlong$

解题代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<algorithm>
using namespace std;
int N,M;
int T,n;
int h[105][105];
struct Matrix{
	long long **x;
	Matrix operator*(const Matrix& t) const{
		Matrix ret;
		for(int i=0;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(x[i][k]+t.x[k][j],ret.x[i][j]);
				}
			}
		}
		return ret;
	} 
	Matrix(){
		x=new long long*[M];
		for(int i=0;i<M;i++){
			x[i]=new long long[M];
		}
		for(int i=0;i<M;i++){
			for(int j=0;j<M;j++){
				x[i][j]=0;
			}
		}
	}
	Matrix(const Matrix&t){
		x=new long long*[M];
		for(int i=0;i<M;i++){
			x[i]=new long long[M];
		}		
		for(int i=0;i<M;i++){
			for(int j=0;j<M;j++){
				x[i][j]=t.x[i][j];
			}
		}
	}
};
Matrix quick_pow(Matrix a,int x){
	Matrix ret;
	while(x){
		if(x&1) ret=ret*a;			
		a=a*a;
		x>>=1;
	}
	return ret;
} 
int main(){
	while(cin>>N>>M) {
		for(int i=1;i<=M;i++){
			for(int j=1;j<=M;j++){
				cin>>h[i][j];
			}
		}
		Matrix trans;
		for(int i=0;i<M;i++){
			for(int j=0;j<M;j++){
				trans.x[i][j]=h[i+1][j+1];
			}
		}
		trans=quick_pow(trans,N-1);
		long long ans=0;
		for(int i=0;i<M;i++){			
			ans=max(ans,trans.x[0][i]);
		}
		cout<<ans<<endl;		
	}
}