不同的路径 II

描述

“不同的路径” 的跟进问题：

现在考虑网格中有障碍物，那样将会有多少条不同的路径？

网格中的障碍和空位置分别用 1 和 0 来表示。

注意事项

m 和 n 均不超过100

样例

[
[0,0,0],
[0,1,0],
[0,0,0]
]

其中，一共有2条不同的路径从左上角到右下角。

思考

1. 相比 不同的路径I 多了一个障碍的条件，即不对有障碍物的点进行操作
2. 由于数组下标长宽弄错，导致下标越界，折腾了半个小时，还是在 VS 里面断点一步一步找出来的，希望大家也可以多注意点

代码

//  By Lentitude


class Solution {
public:
    /**
    * @param obstacleGrid: A list of lists of integers
    * @return: An integer
    */
    int uniquePathsWithObstacles(vector<vector<int> > &obstacleGrid) {
        // write your code here
        vector<vector<int> > vec(obstacleGrid);

        int height = vec.size();
        int width = vec[0].size();

        // 把 0 全部转化为 1, 1 转化为0，便于后面直接对 1 和0分开操作
        for (int i = 0; i != height; ++i) {
            for (int j = 0; j != width; ++j) {
                if (vec[i][j] == 1) {
                    vec[i][j] = 0;
                }
                else {
                    vec[i][j] = 1;
                }
            }
        }

        // 将第一行中 101 -> 100
        for (int i = 0; i != width; ++i) {
            if (vec[0][i] == 0) {
                while (i < width) {
                    vec[0][i++] = 0;
                }
                break;
            }
        }

        /**
        * 将第一列中   1          1
        *            0   ->     0
        *            1          0
        */
        for (int i = 0; i != height; ++i) {
            if (vec[i][0] == 0) {
                while (i < height) {
                    vec[i++][0] = 0;
                }
                break;
            }
        }

        // 如果是单行或者单列，则不需要进行运算，可以直接返回结果
        if (height == 1 || width == 1) {
            return vec[height - 1][width - 1];
        }


        // 否则, 需要进行运算
        for (int i = 1; i != height; ++i) {
            for (int j = 1; j != width; ++j) {
                if (vec[i][j] != 0) {
                    vec[i][j] = vec[i - 1][j] + vec[i][j - 1];
                }
            }
        }

        return vec[height - 1][width - 1];

    }
};