Q老师度假【动态规划dp】【矩阵快速幂优化】

问题描述

忙碌了一个学期的 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 Input

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

Sample Output

2
9

问题分析

• 天数连续，很明显有子结构的性质，可以考虑 DP
设DP状态：
令 f[i][j] 表示第 i 天，穿的衣服为 j 所获得的快乐值总和
根据 DP 状态，不难得到如下的转移方程
• 枚举前一天穿的衣服为 k，即
•f[i][j] = max{ f[i-1][k] + H[k][j] } (1<=k<=M)
直接求解时间复杂度为O（N * M * M）会超时

需要用矩阵快速幂进行优化
矩阵快速幂转化方法

转换过程
•C[i][j] = ∑(A[i][k] ∗ B[k][j]) ⇒ max (A[i][k]+ B[k][j])
• 将累加换成了max，乘法换成了加法
这样是可行的，它依然满足结合律，严谨的数学证明在这里不再展示。

#include<stdio.h>
#include<string.h>
using namespace std;
int M;
long long max(const long long&x,const long long& y){
	if(x>y)return x;
	return y;
}
struct Matrix{
	long long x[110][110];
	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(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 quick_pow(Matrix a,int x){
	Matrix ret;
	memset(ret.x,0,sizeof ret.x);
	while(x){
		if(x&1)ret=ret*a;
		a=a*a;
		x>>=1;
	}
	return ret;
}
int main(){
	int N;
	while(scanf("%d%d",&N,&M)!=EOF){
		Matrix a;
		for(int i=0;i<M;i++)
			for(int j=0;j<M;j++)
				scanf("%lld",&a.x[j][i]);
		Matrix f=quick_pow(a,N-1);
		long long ans=0;
		for(int i=0;i<M;i++)
			for(int j=0;j<M;j++)
				ans=max(ans,f.x[i][j]);
		printf("%lld\n",ans); 
	}
	
	return 0;
}