首先介绍最简单的取数问题:
leetcode64. 最小路径和 https://leetcode-cn.com/problems/minimum-path-sum/
给定一个包含非负整数的 m x n 网格,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。
说明:每次只能向下或者向右移动一步。
示例:
输入:
[
[1,3,1],
[1,5,1],
[4,2,1]
]
输出: 7
解释: 因为路径 1→3→1→1→1 的总和最小。
采用动态规划,维护dp数组,dp[i][j]代表从(0,0)出发到(i,j)的最小路径和,状态转移方程为dp[i][j]=min(dp[i][j-1],dp[i-1][j])+grid[i][j]
class Solution:
def minPathSum(self, grid: List[List[int]]) -> int:
if not grid or not grid[0]:
return 0
m=len(grid)
n=len(grid[0])
dp=[[0 for _ in range(n)] for _ in range(m)]
dp[0][0]=grid[0][0]
for i in range(1,m):
dp[i][0]=dp[i-1][0]+grid[i][0]
for i in range(1,n):
dp[0][i]=dp[0][i-1]+grid[0][i]
for i in range(1,m):
for j in range(1,n):
dp[i][j]=min(dp[i][j-1],dp[i-1][j])+grid[i][j]
return dp[-1][-1]
下面进入正题:方格取数问题
方格取数
有n*n个格子,每个格子里有正数或者0,从最左上角往最右下角走,只能向下和向右,一共走两次(即从左上角走到右下角走两趟),把所有经过的格子的数加起来,求最大值SUM,且两次如果经过同一个格子,则最后总和SUM中该格子的计数只加一次。
其基本思路同上述例子,我们从两条路径同时出发,维护dp[i][i][x][y],代表两个点分别走到(i,j)和(x,y)时的最小路径和,其状态由两个点决定,每个点由前一步的两种状态(top/left)决定,故由2**2=4种状态(top+left,top+top,left+top,left+left)决定那么状态转移方程为:
dp[i][j][k][l] = max(dp[i-1][j][k-1][l],dp[i][j-1][k][l-1]),dp[i-1][j][k][l-1],dp[i][j-1][k-1][l]))+arr[i][j]+ (0 if i==k and j==l else arr[k][l])