Leetcode Algorithm 063. Unique Paths II

Leetcode Algorithm 063. Unique Paths II

Unique Paths II
给定一个m*n的方格，从(0,0)点出发只能向右或向下走，但中间某些格子设了障碍不能走，求走到(m,n)点一共有多少条不同的路径

解题思路

这个题目在 Unique Paths的基础上在某些格子上增加了障碍物，那么意味着到达障碍物的路径其实为0。

假设到(i,j)的 路径有P(i,j)条，用v(i,j) = 1表示该位置有障碍物，用v(i,j) = 0表示该位置没有障碍物，那么
我们可以写出一条动态转移方程：
if v(i,j) = 0, P(i,j) = P(i-1,j) + P(i,j-1)
else if v(i,j) = 1, P(i,j) = 0

初始状态也有一些变换，由于第0行和第0列上的格子只能由一条路径得到，假如前方有一个障碍物的话，后面的格子都不可达，故有：
if v(0,0) = 0, P(0,0) = 1
else if v(0,0) = 1, P(0,0) = 0

P(0,j) = P(0,j-1)
P(i,0) = P(i-1,0)

用题目中3*3的格子，以黑色方格表示障碍物，我们我可以得到：

代码

#include<iostream>
#include<vector>

using namespace std;

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int> >& obstacleGrid) {
        int m = obstacleGrid.size();
        int n = obstacleGrid[0].size();

        int p[m][n];

        p[0][0] = obstacleGrid[0][0] == 0 ? 1 : 0;

        for (int i = 1; i < m; i++) {
            if (obstacleGrid[i][0] == 0)
                p[i][0] = p[i - 1][0];
            else
                p[i][0] = 0;
        }

        for (int j = 1; j < n; j++) {
            if (obstacleGrid[0][j] == 0)
                p[0][j] = p[0][j - 1];
            else
                p[0][j] = 0;

        }

        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (obstacleGrid[i][j] == 0)
                    p[i][j] = p[i - 1][j] + p[i][j - 1];
                else
                    p[i][j] = 0;
            }
        }

        return p[m - 1][n - 1];
    }
};

struct Testcase {
    vector<vector<int> > obstacleGrid;

    void readInput(int m, int n) {
        for (int i = 0; i < m; i++) {
            vector<int> v(n);
            for (int j = 0; j < n; j++) {
                cin >> v[j];
            }

            obstacleGrid.push_back(v);
        }
    }
};

int main() {
    Testcase t;
    t.readInput(3, 3);

    Solution s;

    cout << s.uniquePathsWithObstacles(t.obstacleGrid) << endl;

    return 0;
}

测试样例

0 0 0
0 1 0
0 0 0

输出

2