[动规] [NOIP2007 提高组] 矩阵取数游戏

题目描述

帅帅经常跟同学玩一个矩阵取数游戏：对于一个给定的 $n\times m$ 的矩阵，矩阵中的每个元素 $a_{i,j}$ 均为非负整数。游戏规则如下：

1. 每次取数时须从每行各取走一个元素，共 $n$ 个。经过 $m$ 次后取完矩阵内所有元素；
2. 每次取走的各个元素只能是该元素所在行的行首或行尾；
3. 每次取数都有一个得分值，为每行取数的得分之和，每行取数的得分 = 被取走的元素值 $\times 2^i$，其中 $i$ 表示第 $i$ 次取数（从 $1$ 开始编号）；
4. 游戏结束总得分为 $m$ 次取数得分之和。

帅帅想请你帮忙写一个程序，对于任意矩阵，可以求出取数后的最大得分。

点击查看完整题目信息。

题解

状态转移方程：

$f_{l,r}=max(f_{l+1,r}+v_l\ast 2^{m+l-r},f_{l,r-1}+v_r\ast 2^{m+l-r})$

$f_{l,r}$表示：长度为 m 的一行数字元素里，区间左边第 l 个元素和右边的第 r 个元素之间的最大得分。

题中是$n\ast m$的矩阵，需要对每一行（共 n 行）计算上面的状态转移方程。每一行之间互不影响，帅帅第一次可以取第一行的末尾元素，取第二行的首位元素，两行之间互不影响。

区间 dp，从外层压到里面最小的单个元素，首先初始化dp[1][1]、dp[2][2]、dp[3][3]等，用递归比循环直观。下面是递归代码。

代码（cpp）：

#include<stdio.h>
#include<string.h>

int matrix[81][81] = {0}; // 矩阵
__int128 pow_2[81], dp[81][81][81];

void print(__int128 x)
{
  if(! x) return;
  if(x) print(x / 10);
  putchar(x % 10 + '0');
}

__int128 genDp(int curN, int m, int left, int right)
{
  __int128 leftDp, rightDp;
  if (dp[curN][left][right] != -1)
    return dp[curN][left][right];
  if (left < right)
  {
    leftDp = genDp(curN, m, left + 1, right) + matrix[curN][left] * pow_2[m + left - right];
    rightDp = genDp(curN, m, left, right - 1) + matrix[curN][right] * pow_2[m + left - right];
    dp[curN][left][right] = leftDp > rightDp ? leftDp : rightDp;
  }
  return dp[curN][left][right];
}

int main()
{
  int n, m;
  
  int mVal;
  __int128 ans = 0;
  int i, j;
  memset(dp, -1, sizeof(dp));

  scanf("%d %d", &n, &m);
  for (i = 0; i < n; ++ i)
    for (j = 0; j < m; ++ j)
    {
      scanf("%d", &mVal);
      matrix[i][j] = mVal;
    }

  pow_2[0] = 1;
  for (i = 1; i < m + 1; ++ i)
    pow_2[i] = pow_2[i - 1] * 2;

  for (i = 0; i < n; ++ i)
  {
    for (j = 0; j < m; ++ j)
      dp[i][j][j] = matrix[i][j] * pow_2[m]; // 初始化 dp[j][j]，dp[j][j] 表示第 j 个元素是最后一个取到的元素（第 m 次取数），所以乘 2 的 m 次方
    ans += genDp(i, m, 0, m - 1);
  }
  if (! ans)
    printf("0");
  else
    print(ans);
  return 0;
}