LeetCode: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.


典型的動態規划問題。

設dp[i][j]表示從左上角到grid[i][j]的最小路徑和。那么dp[i][j] = grid[i][j] + min( dp[i-1][j], dp[i][j-1] );

下面的代碼中,為了處理計算第一行和第一列的邊界條件,我們令dp[i][j]表示從左上角到grid[i-1][j-1]的最小路徑和,最后dp[m][n]是我們所求的結果

class Solution {
public:
    int minPathSum(vector<vector<int> > &grid) {
        int row = grid.size(),col;
        if(row == 0)return 0;
        else col = grid[0].size();
        vector<vector<int> >dp(row+1, vector<int>(col+1, INT_MAX));
        dp[0][1] = 0;
        for(int i = 1; i <= row; i++)
            for(int j = 1; j <= col; j++)
                dp[i][j] = grid[i-1][j-1] + min(dp[i][j-1], dp[i-1][j]);
        return dp[row][col];
    }
};

 

注意到上面的代碼中dp[i][j] 只和上一行的dp[i-1][j]和上一列的dp[i][j-1]有關,因此可以優化空間為O(n)(准確來講空間復雜度可以是O(min(row,col

)))                          本文地址

class Solution {
public:
    int minPathSum(vector<vector<int> > &grid) {
        int row = grid.size(),col;
        if(row == 0)return 0;
        else col = grid[0].size();
        vector<int >dp(col+1, INT_MAX);
        dp[1] = 0;
        for(int i = 1; i <= row; i++)
            for(int j = 1; j <= col; j++)
                dp[j] = grid[i-1][j-1] + min(dp[j], dp[j-1]);
        return dp[col];
    }
};

問題擴展

最大路徑和只需要把上面的遞推公式中的min換成max。

現在有個問題,如果兩個人同時從左上角出發,目的地是右下角,兩個人的路線不能相交(即除了出發點和終點外,兩個人不同通過同一個格子),使得兩條路徑的和最大。(這和一個人先從左上角到右下角,再回到左上角是相同的問題)。

這是雙線程動態規划問題:假設網格為grid,dp[k][i][j]表示兩個人都走了k步,第一個人向右走了i步,第二個人向右走了j步 時的最大路徑和(只需要三個變量就可以定位出兩個人的位置grid[k-i][i-1] 、 grid[k-j][j-1]),那么

dp[k][i][j] = max(dp[k-1][i-1][j-1], dp[k-1][i][j], dp[k-1][i-1][j], dp[k-1][i][j-1]) + grid[k-i][i-1] + grid[k-j][j-1]  (我們假設在起始位置時就已經走了一步)

 

這個方程的意思是從第k-1步到第k步,可以兩個人都向右走、都向下走、第一個向下第二個向右、第一個向右第二個向下,這四種走法中選擇上一步中路徑和最大的。

 

由於要保證兩條路線不能相交,即兩個人走的過程中,有一個人向右走的步數永遠大於另一個人向右走的步數,我們不妨設第二個人向右走的步數較大,即dp[k][i][j]中j > i才是有效的狀態。走到終點的步數為:網格的行數+網格的列數-1

 

需要注意的是:當走了k步時,某個人向右走的步數必須 > k - 網格的行數,如果向右走的步數 <= k-行數,那么向下走的步數 = k-向右走的步數 >= 行數,此時超出了網格的范圍。由於我們假設了 j > i,因此只要保證 i > k-網格行數即可。

代碼如下:

int max2PathSum(vector<vector<int> > grid)
{
    int row = grid.size(), col = grid[0].size();
    vector<vector<vector<int> > > dp(row+col, vector<vector<int> >(col+1, vector<int>(col+1, 0)));
    for(int step = 2; step <= row+col-2; step++)
        for(int i = max(1, step-row+1); i <= step && i <= col; i++)
            for(int j = i+1; j <= step && j <= col; j++)
            {
                
                dp[step][i][j] = max(max(dp[step-1][i][j], dp[step-1][i-1][j-1]), max(dp[step-1][i-1][j], dp[step-1][i][j-1]));
                dp[step][i][j] += (grid[step-i][i-1] + grid[step-j][j-1]);
            }
    return dp[row+col-2][col-1][col] + 2*grid[row-1][col-1] + 2*grid[0][0];
}

 

我們最終的目標是dp[row+col-1][col][col] = max{dp[row+col-2][col-1][col-1], dp[row+col-2][col][col], dp[row+col-2][col-1][col], dp[row+col-2][col][col-1]} + 2*grid[row-1][col-1]

由於dp[row+col-2][col-1][col-1], dp[row+col-2][col][col], dp[row+col-2][col][col-1]都是無效的狀態(dp[k][i][j]中j > i才是有效的狀態),

所以dp[row+col-1][col][col]  = dp[row+col-2][col-1][col] + 2*grid[row-1][col-1],代碼中最后結果還加上了所在起點的的網格值。

由以上可知,循環中最多只需要求到了dp[row+col-2][][]。

 

nyoj中 傳紙條(一)就是這個問題,可以在這一題中測試上述函數的正確性,測試代碼如下:

int main()
{
    int n;
    scanf("%d",&n);
    for(int i = 1; i <= n; i++)
    {
        int row, col;
        scanf("%d%d", &row, &col);
        vector<vector<int> >grid(row, vector<int>(col));
        for(int a = 0; a < row; a++)
            for(int b = 0; b < col; b++)
                scanf("%d", &grid[a][b]);
        printf("%d\n", max2PathSum(grid));
    }
    return 0;
}

 

這個問題還可以使用最小費用流來解決,具體可以參考here

 

 

【版權聲明】轉載請注明出處http://www.cnblogs.com/TenosDoIt/p/3774804.html


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM