63. Unique Paths II

来源:互联网 发布:知其雄,守其雌哲学道理 编辑:程序博客网 时间:2024/06/17 16:57

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.

比较容易想起的方法是递归寻找,每个分支向下或者向右(有点像二叉树),遇到1代表无法通过,如果 到终点则数量加一,可惜这种方式递归调用的层次太深,查找量很大,会超时。
实现方式:

class Solution {public:    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {        if (obstacleGrid.size() == 0 || obstacleGrid[0].size() == 0)        {            return 0;        }        int result = 0;        if (obstacleGrid[0][0] != 1)        {            find_way(obstacleGrid,0,0,result);        }        return result;    }    void find_way(vector<vector<int>>& obstacleGrid,int m,int n,int &result){        if (m == obstacleGrid.size() - 1 && n == obstacleGrid[0].size() - 1 && obstacleGrid[m][n] != 1)        {            result++;            return;        }        if (n + 1 < obstacleGrid[0].size() && obstacleGrid[m][n + 1] != 1)        {            find_way(obstacleGrid,m,n + 1,result);        }        if (m + 1 < obstacleGrid.size() && obstacleGrid[m + 1][n] != 1)        {            find_way(obstacleGrid,m + 1,n,result);        }    }};

考虑四4*5的方块

– – – – – 0 0 0 0 0 0 0 0 1 0 0 0 1 0 0 0 0 0 0 0

第一块如果可达,则只有一种方式。
到达各个点的方式的数量。
(2,2)

– – – – – 1 1 1 1 1 1 2 1 1

(3,3)

– – – – – 1 1 1 1 1 1 2 3 1 3 0 1

实际上到达一个方块的所有方式等于它上边与右边到达方式的和。
那么可以用递推的方式计算出所有块到达的方式,需要一个保存所有达到方式的二维数组。
实现:

class Solution {public:    int uniquePathsWithObstacles(vector<vector<int>>& obstacleGrid) {        //先去除非正常情形        if (obstacleGrid.size() == 0 || obstacleGrid[0].size() == 0 || obstacleGrid[0][0] == 1 ||             obstacleGrid[obstacleGrid.size() - 1][obstacleGrid[0].size() - 1] == 1)        {            return 0;        }        int height = obstacleGrid.size();        int width = obstacleGrid[0].size();        vector<vector<int>> assis_array(height,vector<int>(width,0));       //全部为0的二维数组        assis_array[0][0] = 1;//第一个点如果可达,到达方式为数目为1        for (int i = 0; i < height; ++i)        {            for (int j = 0; j < width; ++j)            {                if (i == 0 && j == 0) continue;//我们已经设置过了                if(obstacleGrid[i][j] == 1) continue;                int left = 0;                int up = 0;                if (j > 0)                {                    left = assis_array[i][j - 1];                }                if (i > 0)                {                    up = assis_array[i - 1][j];                }                assis_array[i][j] = left + up;            }        }        return assis_array[height - 1][width - 1];    }};
0 0
原创粉丝点击