Week14—度假衬衫选择

问题描述

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

问题分析

本题是DP问题与矩阵快速幂的结合，天数连续，很明显有子结构的性质，可以考虑 DP，n 不是很大，因此不一定是矩阵快速幂优化，先列出 DP状态和 DP 转移方程再进行判断；题目问 n 天后，最多快乐值，且每天的快乐值由今日与昨日两天的衣服决定
• 因此令 f[i][j] 表示第 i 天，穿的衣服为 j 所获得的快乐
值总和,枚举前一天穿的衣服为 k，即f[i][j] = max (f[i-1][k]+ H[k][jj]
输出答案 max{dp[i][j]}(0⩽i,j⩽M−1)，复杂度9次方，需要优化：
有关矩阵快速幂基本知识

代码

#include <iostream>
#include <string.h>
using namespace std;

int m,n;
struct Matrix{
	long long x[110][110];
	
	Matrix operator*(Matrix&b)
	{
		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]+b.x[k][j]);
			}
		}
		return ret;
	}
	Matrix()
	{
		memset(x,0,sizeof(x));
	}
	Matrix(Matrix&t)
	{
		memcpy(x,t.x,sizeof(x));
	}
}; 
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) 
	{
		Matrix temp;
		for(int i=0;i<m;i++)
			for(int j=0;j<m;j++)
				cin>>temp.x[i][j];
		temp=quick_pow(temp,n-1);
		long long ans=0;
		for(int i=0;i<m;i++)
			for(int j=0;j<m;j++)
				ans=max(ans,temp.x[i][j]);
		cout<<ans<<endl;
	}
}