741. 摘樱桃
摘樱桃
题目描述:
一个N x N的网格(grid) 代表了一块樱桃地,每个格子由以下三种数字的一种来表示:
0 表示这个格子是空的,所以你可以穿过它。
1 表示这个格子里装着一个樱桃,你可以摘到樱桃然后穿过它。
-1 表示这个格子里有荆棘,挡着你的路。
你的任务是在遵守下列规则的情况下,尽可能的摘到最多樱桃:
从位置 (0, 0) 出发,最后到达 (N-1, N-1) ,只能向下或向右走,并且只能穿越有效的格子(即只可以穿过值为0或者1的格子);
当到达 (N-1, N-1) 后,你要继续走,直到返回到 (0, 0) ,只能向上或向左走,并且只能穿越有效的格子;
当你经过一个格子且这个格子包含一个樱桃时,你将摘到樱桃并且这个格子会变成空的(值变为0);
如果在 (0, 0) 和 (N-1, N-1) 之间不存在一条可经过的路径,则没有任何一个樱桃能被摘到。
示例 1:
输入: grid =
[[0, 1, -1],
[1, 0, -1],
[1, 1, 1]]
输出: 5
解释:
玩家从(0,0)点出发,经过了向下走,向下走,向右走,向右走,到达了点(2, 2)。
在这趟单程中,总共摘到了4颗樱桃,矩阵变成了[[0,1,-1],[0,0,-1],[0,0,0]]。
接着,这名玩家向左走,向上走,向上走,向左走,返回了起始点,又摘到了1颗樱桃。
在旅程中,总共摘到了5颗樱桃,这是可以摘到的最大值了。
说明:
grid 是一个 N * N 的二维数组,N的取值范围是1 <= N <= 50。
每一个 grid[i][j] 都是集合 {-1, 0, 1}其中的一个数。
可以保证起点 grid[0][0] 和终点 grid[N-1][N-1] 的值都不会是 -1。
思路:
时间复杂度:O(n^3),空间复杂度O(n^3)
是我遇到的线性dp里面最难的题目,需要三维考虑,x1+y1 = x2+y2 = k
代码:
func cherryPickup(grid [][]int) int {
n := len(grid)
dp := make([][][]int, n*2 - 1)
for i:=0;i< n *2-1;i++{
dp[i] = make([][]int, n)
for j:=0;j < n; j++{
dp[i][j] = make([]int, n)
for k := range dp[i][j] {
dp[i][j][k] = math.MinInt32
}
}
}
dp[0][0][0] = grid[0][0]
for k:=1; k< 2*n - 1; k++{
for x1 := max(0, k - n +1); x1 <=min(n-1, k);x1++{
y1 := k - x1
if grid[x1][y1] == -1{
continue
}
for x2 := x1; x2 <=min(n-1,k); x2++{
y2 := k - x2
if grid[x2][y2] == -1{
continue
}
res := dp[k-1][x1][x2]
if x1 >0{
res = max(res, dp[k-1][x1-1][x2])
}
if x2 >0{
res = max(res, dp[k-1][x1][x2-1])
}
if x1 >0 && x2 >0{
res = max(res, dp[k-1][x1-1][x2-1])
}
res += grid[x1][y1]
if x1 != x2{
res += grid[x2][y2]
}
dp[k][x1][x2] = res
}
}
}
//fmt.Println(dp)
return max(dp[2*n-2][n-1][n-1], 0)
}
func max(a,b int)int{
if a>b{
return a
}
return b
}
func min(a,b int)int{
if a>b{
return b
}
return a
}
代码效率:
执行用时:16 ms, 在所有 Go 提交中击败了71.72%的用户
内存消耗:7.3 MB, 在所有 Go 提交中击败了50.94%的用户
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!