Follow up for "Unique Paths":
Now consider if some obstacles are added to the grids. How many unique paths would there be?
An obstacle and empty space is marked as 1
and 0
respectively
in the grid.
For example,
There is one obstacle in the middle of a 3x3 grid as illustrated below.
[ [0,0,0], [0,1,0], [0,0,0] ]
The total number of unique paths is 2
.
class Solution: # @param obstacleGrid, a list of lists of integers # @return an integer def uniquePathsWithObstacles(self, obstacleGrid): if not obstacleGrid: return 0 if obstacleGrid[0][0]== 0: obstacleGrid[0][0]=1 else: return 0 m,n =len(obstacleGrid),len(obstacleGrid[0]) for j in xrange(0,n): for i in xrange(0,m): if obstacleGrid[i][j] == 1 and (i != 0 or j != 0): obstacleGrid[i][j] = 0 elif i == 0 and j == 0: continue elif i == 0: obstacleGrid[i][j] = obstacleGrid[i][j-1] elif j == 0: obstacleGrid[i][j] = obstacleGrid[i-1][j] else: obstacleGrid[i][j] = obstacleGrid[i-1][j]+obstacleGrid[i][j-1] return obstacleGrid[m-1][n-1]
原文地址:http://blog.csdn.net/u010006643/article/details/44836193