NC16645 矩阵取数游戏(记忆化搜索)

题目链接

题意：
$有n行m列的矩阵$
$对于第i行，可以取出行首或行尾的一个元素$
$对于每一行第k次取出可以得到2^k\ast a_{ij}的分数$
$问最多能得到多少分$
题解：
$n,m<=80,a_{ij}<=8000$
$可以发现，每一行是独立的，所以对每一行进行单独分析$
$使用记忆化搜索，f[i][j]表示取走i到j两边的元素，最多能得到的分$
$每一行有m个元素，所以可以计算出当前已经取元素的次数$
$m-r+l就是取元素的次数，可以用来计算2^k$
$然后列出转移方程式$
$f[i][j]=f[i+1][r]\ast a_i\ast 2^k+f[i][r-1]\ast a_r\ast 2^k$
$边界就是i==j的时候，说明这是最后一个元素$
$就可以直接返回a_i\ast 2^m$
$但是需要注意的是，这个数据是会爆longlong的$
$所以我直接用的python$
AC代码

f = [[0 for i in range(100)] for i in range(100)]
n, m = map(int, input().split())
a = [0 for i in range(100)]
def dfs(l, r) :
    len = r - l + 1
    x = m - len + 1
    if f[l][r] :
        return f[l][r]
    if l == r :
        return a[l] * (2 ** x)
    f[l][r] = max(dfs(l + 1, r) + a[l] * 2 ** x, dfs(l, r - 1) + a[r] * 2 ** x)
    return f[l][r]

ans = 0
for i in range(n):
     f = [[0 for i in range(100)] for i in range(100)]
     a = list(map(int, input().split()))
     ans += dfs(0, m - 1)
print(ans)