Q老师度假

Description

忙碌了一个学期的 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[i][j]为前i天假期，最后一天穿第j件衬衫可以获得的快乐值。
初始化 dp[1][j]=0(0⩽j⩽M−1)。

转移过程：

dp[i][j]=max{dp[i−1][k]+f[k][j]}(1⩽k⩽M);

输出答案 max{dp[i][j]}(0⩽i,j⩽M−1).
但是:
所以需要优化:

代码：

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxm = 105;
int n, M;

struct Matrix
{
    long long m[maxm][maxm];

    Matrix() {}

    Matrix(const Matrix &t)
    {
        memcpy(m, t.m, sizeof(m));
    }

    Matrix operator*(const Matrix &t) const
    {
        Matrix res;
        for (int i = 1; i <= M; i++)
        {
            for (int j = 1; j <= M; j++)
            {
                res.m[i][j] = 0;
                for (int k = 1; k <= M; k++)
                {
                    res.m[i][j] = max(res.m[i][j], m[i][k] + t.m[k][j]);   //新定义的乘法
                }
            }
        }
        return res;
    }
};
Matrix quick_pow(Matrix a, int x)
{
    Matrix res;
    memset(res.m,0,sizeof(res.m));
    while (x)
    {
        if (x & 1)
        {
            res = res * a;
        }
        a = a * a;
        x >>= 1;
    }
    return res;
}

Matrix A, f;
int main()
{
    while (~scanf("%d %d", &n, &M))
    {
        for (int i = 1; i <= M; i++)
        {
            for (int j = 1; j <= M; j++)
            {
                scanf("%lld", &A.m[i][j]);
            }
        }
        f = quick_pow(A, n-1 );   //快速幂
        long long ans=0;
        for(int i=1;i<=M;i++)
        {
            for(int j=1;j<=M;j++)
                ans=max(ans,f.m[i][j]);    //最大值为结果
        }
        printf("%lld\n", ans);
    }

    return 0;
}