【Leetcode】741. Cherry Pickup 741. 摘樱桃

【Leetcode】741. Cherry Pickup 741. 摘樱桃_第1张图片
【Leetcode】741. Cherry Pickup 741. 摘樱桃_第2张图片

解法

总而言之是DP的解法
把问题转换成有2个人从左上向右下走,一共能摘到的樱桃的最大数量
每个人都要走m+n-2步,走第t步时的状态可以表示为(t,r1,c1,r2,c2)
当然,这个五元组是冗余的,只需要三维就可以表示这个向量了,因为:
t = r1+c1=r2+c2
在每个时刻,每个人都可以向下或向右,所以总共会产生4种情况:

  • p1向下,p2向下:(t+1,r1+1,c1,r2+1,c2)
  • p1向下,p2向右:(t+1,r1+1,c1,r2,c2+1)
  • p1向右,p2向下:(t+1,r1,c1+1,r2+1,c2)
  • p1向右,p2向右:(t+1,r1,c1+1,r2,c2+1)

解法一:自顶向下的DP,记忆化搜索

状态(t,r1,c1,r2,c2)的值代表从该状态到终点(m+n-2,m-1,n-1,m-1,n-1)摘到的樱桃的个数
要注意处理这个状态无法到达终点的情况
这里我们提取出(r1,c1,c2)作为状态的表示

class Solution(object):
    def cherryPickup(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        m = len(grid)
        n = len(grid[0])
        from collections import defaultdict
        mem = defaultdict(lambda:defaultdict(lambda :defaultdict(int)))
        mem[m-1][n-1][n-1] = grid[-1][-1]

        def isable(i, j, k):
            c = i + j - k
            return i >= 0 and i < m and j >= 0 and j < n and k >= 0 and k < n and c >= 0 and c < m and grid[i][
                j] != -1 and grid[c][k] != -1

        def solve(r1, c1, c2):
            if r1 not in mem or c1 not in mem[r1] or c2 not in mem[r1][c1]:
                res = -1
                if isable(r1 + 1, c1, c2):
                    res = max(res, solve(r1 + 1, c1, c2))
                if isable(r1 + 1, c1, c2 + 1):
                    res = max(res, solve(r1 + 1, c1, c2 + 1))
                if isable(r1, c1 + 1, c2 + 1):
                    res = max(res, solve(r1, c1 + 1, c2 + 1))
                if isable(r1, c1 + 1, c2):
                    res = max(res, solve(r1, c1 + 1, c2))
                if res==-1:
                    mem[r1][c1][c2] = -1
                else:
                    if grid[r1][c1] == 1:
                        res += 1
                    if c1 != c2 and grid[r1 + c1 - c2][c2] == 1:
                        res += 1
                    mem[r1][c1][c2] = res
            return mem[r1][c1][c2]

        res = solve(0, 0, 0)
        return res if res != -1 else 0

解法二:向底向上的DP

自底身上的DP有可能会遍历许多不可能的情况,需要在循环里就跳过
一个比较好的省空间的方案是取(t,r1,r2)表示状态,这样状态t只会与状态t-1有关
利用背包问题省空间的原理就可以把记录缩小到2维

class Solution(object):
    def cherryPickup(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        m = len(grid)
        n = len(grid[0])
        from collections import defaultdict
        mem = defaultdict(lambda:defaultdict(int))

        def isable(t, i,c):
            j,k = t-i,t-c
            return i >= 0 and i < m and j >= 0 and j < n and k >= 0 and k < n and c >= 0 and c < m and grid[i][
                j] != -1 and grid[c][k] != -1

        for t in xrange(m+n-1):
            for r1 in xrange(min(m,t+1),-1,-1):
                for r2 in xrange(min(m,t+1),-1,-1):
                    if isable(t,r1,r2):
                        res = -1
                        if isable(t-1,r1,r2):
                            res = max(res, mem[r1][r2])
                        if isable(t-1, r1-1,r2):
                            res = max(res, mem[r1-1][r2])
                        if isable(t-1,r1,r2-1):
                            res = max(res, mem[r1][r2-1])
                        if isable(t-1,r1-1,r2-1):
                            res = max(res, mem[r1-1][r2-1])
                        if res == -1 and t!=0:
                            mem[r1][r2] = -1
                        else:
                            if res ==-1:
                                res = 0
                            if grid[r1][t-r1] == 1:
                                res += 1
                            if r1 != r2 and grid[r2][t - r2] == 1:
                                res += 1
                            mem[r1][r2] = res
                    else:
                        mem[r1][r2] = -1
        return mem[m-1][m-1] if mem[m-1][m-1]!=-1 else 0

你可能感兴趣的:(Leetcode)