Unique Paths
题目大意
机器人从起点到终点有多少条不同的路径,只能向右或者向下走。
解题思路
动态规划
由于只能有向下向右,只有从[1][1]开始的格子需要选择走法,第一行和第一列所有都只有一种走法,所有都设置成1,(这里图方便所有都初始化为1),然后循环计算出所有其他的。
dp[j][i] = dp[j - 1][i] + dp[j][i - 1]
代码
class Solution(object):
def uniquePaths(self, m, n):
""" :type m: int :type n: int :rtype: int """
dp = [[1 for __ in range(n)] for __ in range(m)]
# print dp
for i in range(1, n):
for j in range(1, m):
dp[j][i] = dp[j - 1][i] + dp[j][i - 1]
return dp[m - 1][n - 1]
Unique Paths II
题目大意
如果道路上有障碍,机器人从起点到终点有多少条不同的路径,只能向右或者向下走。0表示道路通行,1表示有障碍。
解题思路
和上题区别仅仅是有障碍,那么有障碍的地方就是0(不可能走到),然后初始化一下第一行和第一列(若其中有障碍则之后皆为0),最后还是上题那样做。
dp[j][i] = dp[j - 1][i] + dp[j][i - 1] (s[i][j] != 1) # 无障碍
dp[j][i] = 0 (s[i][j] = 1) # 有障碍
代码
class Solution(object):
def uniquePathsWithObstacles(self, obstacleGrid):
""" :type obstacleGrid: List[List[int]] :rtype: int """
if obstacleGrid[0][0] == 1: # 如果第一个格子有障碍,就无法走
return 0
m = len(obstacleGrid) # 纵向
n = len(obstacleGrid[0]) # 横向
dp = [[0 for __ in range(n)] for __ in range(m)] # 初始化,所有为0
# 第一个可以走设置为1
dp[0][0] = 1
# 检查第一行和第一列是否有障碍
for i in range(1, m):
dp[i][0] = dp[i - 1][0] if obstacleGrid[i][0] == 0 else 0
for j in range(1, n):
dp[0][j] = dp[0][j - 1] if obstacleGrid[0][j] == 0 else 0
for i in range(1, m):
for j in range(1, n):
if obstacleGrid[i][j] == 1:
dp[i][j] = 0
else:
dp[i][j] = dp[i - 1][j] + dp[i][j - 1]
return dp[m - 1][n - 1]