题解 | #挤奶路径#

104 阅读2分钟

描述

一位农夫在一个m x n的农场里养了一些奶牛。农夫每次只能向下或向右移动一步去挤奶。但是农场中有些地方有障碍物,农夫不能到达那些位置。现在农夫想知道,他从农场的左上角出发,有多少种不同的路径可以到达农场的右下角。 农场中的障碍物和空位置分别用1和0表示。

示例1

输入:[[0,0,0],[0,1,0],[0,0,0]]
返回值:2
说明:3x3的农场中有一个障碍物。
从左上角到右下角一共有2条不同的路径:

向右 -> 向右 -> 向下 -> 向下向下 -> 向下 -> 向右 -> 向右

示例2

输入:[[0,1],[0,0]]
返回值:1

备注:

m == cows.length
n == cows[i].length
1 <= m, n <= 50

cows[i][j] 为 0 或 1

知识点

动态规划

解题思路

定义一个二维数组dp,dp[i][j]表示从农场的左上角到达位置(i, j)的不同路径数。

首先,我们初始化dp数组。如果农场的左上角有障碍物(即cows[0][0]为1),则路径数为0;否则,路径数为1,即dp[0][0] = 1。

然后,我们根据动态规划的递推关系式来更新dp数组。对于位置(i, j),如果当前位置为障碍物,则路径数为0,即dp[i][j] = 0。否则,可以根据上方位置和左方位置的路径数来计算当前位置的路径数,即dp[i][j] = dp[i-1][j] + dp[i][j-1]。

最后的dp[m-1][n-1]就是最终ans。

Java题解

import java.util.*;
public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     * @param cows int整型二维数组
     * @return int整型
     */
    public int uniquePathsWithObstacles (int[][] cows) {
        // write code here
        int m = cows.length;
        int n = cows[0].length;

        int[][] dp = new int[m][n];

        // 初始化第一个位置
        if (cows[0][0] == 0) {
            dp[0][0] = 1;
        } else {
            dp[0][0] = 0;
        }

        // 初始化第一列
        for (int i = 1; i < m; i++) {
            if (cows[i][0] == 0) {
                dp[i][0] = dp[i - 1][0];
            } else {
                dp[i][0] = 0;
            }
        }
        // 初始化第一行
        for (int j = 1; j < n; j++) {
            if (cows[0][j] == 0) {
                dp[0][j] = dp[0][j - 1];
            } else {
                dp[0][j] = 0;
            }
        }

        // 更新dp数组
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (cows[i][j] == 0) {
                    dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
                } else {
                    dp[i][j] = 0;
                }
            }
        }
        return dp[m - 1][n - 1];
    }
}