码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] Unique Paths && Unique Paths II && Minimum Path Sum (动态规划之 Matrix DP )

时间:2015-01-11 21:30:24      阅读:319      评论:0      收藏:0      [点我收藏+]

标签:

Unique Paths

https://oj.leetcode.com/problems/unique-paths/

A robot is located at the top-left corner of a m x n grid (marked ‘Start‘ in the diagram below).

The robot can only move either down or right at any point in time. The robot is trying to reach the bottom-right corner of the grid (marked ‘Finish‘ in the diagram below).

How many possible unique paths are there?

技术分享

Above is a 3 x 7 grid. How many possible unique paths are there?

Note: m and n will be at most 100.

我的思路

我个人通常喜欢将此类问题,叫做 Matrix DP 问题,这道题很显然需要使用动态规划来做,一般做此类 Matrix DP 问题,通常需要注意以下几点:

state: f[x][y] 表示我从起点走到坐标x,y……
function: 研究最后一步怎么走
intialize: 起点
answer: 终点

先来看 state ,可以这样定义,f[x][y] 表示从起点 (0,0) 出发,到达 (x,y) 的 unique paths

再来看看怎么找状态转移方程,不难发现,从起点出发到达点 (x,y) 的 unique paths 实际上等于从起点出发到达该点左边一点(也就是(x,y-1))的 unique paths 加上从起点出发到达该点上面一点(也就是(x-1, y))的 unique paths 后的和。

显然该 f 矩阵的初始化是将最左边的一列和最上面的一行置为 1 ,因为由起点出发无论往下走还是往右走,该列或者该行上的每一点的 unique paths 均为1,注意即使 start 位置等于 finish 位置(也就是只有一个点),其 unique paths 也为1(自己走向自己算 1 )。

下面是我在 LeetCode 上 AC 的代码:

/**
 * Zhou J
 */ 

class Solution 
{
public:
    int uniquePaths(int m, int n) 
    {
        if (m == 0 || n == 0) {
            return 0;
        }
        
        int sum[m][n];
        
        for (int i = 0; i < m; i++) {
            sum[i][0] = 1;
        }
        
        for (int i = 0; i < n; i++) {
            sum[0][i] = 1;
        }
        
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                sum[i][j] = sum[i - 1][j] + sum[i][j - 1];
            }
        }
        
        return sum[m - 1][n - 1];
    }
};

如果你们理解了上面这一题,可以接着来看下面的 follow question。

Follow Question

Unique Paths II

https://oj.leetcode.com/problems/unique-paths-ii/

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.

Note: m and n will be at most 100.

下面是我 AC 的代码:

/**
 * Zhou J
 */ 

class Solution {
public:
    int uniquePathsWithObstacles(vector<vector<int> > &obstacleGrid) {
        if (obstacleGrid.size() == 0)
        {
            return 0;
        }
        
        int numOfRows = obstacleGrid.size();
        int numOfCols = obstacleGrid[0].size();
        vector<vector<int>> sum(obstacleGrid);
        
        // initialize the left column
        for (size_t ix = 0; ix != numOfRows; ++ix)
        {
            if (obstacleGrid[ix][0] == 0)
            {
                sum[ix][0] = 1;
            }
            else
            {
                for (size_t newIx = ix; newIx != numOfRows; ++newIx)
                {
                    sum[newIx][0] = 0;
                }
                break;
                     
        }
        
        // initialize the top row
        for (size_t ix = 0; ix != numOfCols; ++ix)
        {
            if (obstacleGrid[0][ix]  == 0)
            {
                sum[0][ix] = 1;
            }
            else
            {
                for (size_t newIx = ix; newIx != numOfCols; ++newIx)
                {
                    sum[0][newIx] = 0;
                }
                break;
            }
        }
        
        // switch the state
        for (size_t i = 1; i != numOfRows; ++i)
        {
            for (size_t j = 1; j != numOfCols; ++j)
            {
                if (obstacleGrid[i][j] == 0)
                {
                    sum[i][j] = sum[i-1][j] + sum[i][j-1];
                }
                else
                {
                    sum[i][j] = 0;
                }
                
            }
        }
        
        // return the answer
        return sum[numOfRows - 1][numOfCols - 1];
    }
};

这倒题目相对上一题,有这样两个注意点:

1. 在初始化行和列的时候,如果有某一个点为障碍物,那么不仅仅该位置所对应的 sum 将置为0,而是其接下来的每一个位置所对应的 sum 都要置为0,这很好理解,一行或一列都属于直线,其中只要有一个障碍物,那么这一整条线路就废掉了。

2. 状态转移的过程中,如果遇到障碍物,改点应该直接返回 0 。

Minimum Path Sum

我们最后再来看一道 Matrix DP 相关的问题,如下:

Minimum Path Sum

https://oj.leetcode.com/problems/minimum-path-sum/

Given a m x n grid filled with non-negative numbers, find a path from top left to bottom right which minimizes the sum of all numbers along its path.

Note: You can only move either down or right at any point in time.

此题,实际上和上面两题属于一个套路,只不过此处的状态矩阵 sum 存放的是最短路径和。从起点出发到达某个位置的最短路径和,一定是从起点出发到达该位置左边或者上面一点的最短路径(两者取较小的)加上该点自身的路径。

state: f[x][y]从起点走到x,y的最短路径
function: f[x][y] = min(f[x-1][y], f[x][y-1]) + cost[x][y]
intialize: f[0][0] = cost[0][0]
             f[i][0] = sum(0,0 -> i,0)
             f[0][i] = sum(0,0 -> 0,i)
answer: f[n-1][m-1]

下面是 AC 的代码:

/**
 * Zhou J
 */
class Solution {
public:
    int minPathSum(vector<vector<int> > &grid) 
    {
        if (grid.size() == 0)
        {
            return 0;
        }
        
        int numOfRows = grid.size();
        int numOfCols = grid[0].size();
        vector<vector<int>> sum(grid);
        
        sum[0][0] = grid[0][0];
        
        // initialize the left column
        for (size_t ix = 1; ix != numOfRows; ++ix)
        {
            sum[ix][0] = sum[ix - 1][0] + grid[ix][0];
        }
        
        // initialize the top row
        for (size_t ix = 1; ix != numOfCols; ++ix)
        {
            sum[0][ix] = sum[0][ix - 1] + grid[0][ix];
        }
        
        // switch the state
        for (size_t i = 1; i != numOfRows; ++i)
        {
            for (size_t j = 1; j != numOfCols; ++j)
            {
                sum[i][j] = min(sum[i][j - 1], sum[i - 1][j]) + grid[i][j];
            }
        }
        
        // return the minimum path sum
        return sum[numOfRows - 1][numOfCols - 1];
        
    }
};

刷题心得

LeetCode 是一个非常好的平台,我们在上面做题时,一定要自己先思考,不要急于去 google 答案,如果你的代码 AC 不过,LeetCode 是会把过不了的 TestCase 显示给你的,根据这些提示,我们一步步修正答案,往往会得到很好的锻炼。

个人以为 Matrix DP 一类的动态规划问题通常是最简单的,对于此类题目,initialize 这一步往往是初始化起点对应的行和列。在之后的文章中,我也会针对北美一些面试题中涉及到的动态规划问题做进一步分析。

笔者目前还在国内读硕士,但是前些阶段和一些目前仍然在国内工作,但是日后想去湾区工作的朋友们交流,有一个问题是如果经常跳槽的话,会不会影响北美的面试?答案是会有影响的。我咨询了一个Facebook 的面试官,他的意思是做 intern 经常换工作是无所谓的,但是 fulltime 经常换的话,影响还是很大的。打个比方来说,intern 就相当于女朋友,这个社会现在还是允许我们经常换女朋友的(是有点贱贱嗒),但是 fulltime 就相当于 wife ,你总不能老离婚吧?!

PS:我是有多想去湾区工作啊,求内推~!!

[LeetCode] Unique Paths && Unique Paths II && Minimum Path Sum (动态规划之 Matrix DP )

标签:

原文地址:http://www.cnblogs.com/jianxinzhou/p/4217005.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!