Minimum Path Sum (Java)

来源:互联网 发布:panabit官网端口 编辑:程序博客网 时间:2024/06/06 00:24

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 - 1][j]和dp[i][j-1],由于for循环是从1开始的,所以要对dp数组的上边界和左边界赋值。

如果不想特别处理边界就将dp元素初始化为Integer.MAX_VALUE,然后将随便dp[1][0]或dp[0][1]置为0,转移方程改为dp[i][j] = grid[i - 1][j - 1] + min{dp[i - 1][j], dp[i][j-1]}即可。

还有更简单的方法能达到O(n),再写的时候注意下。

Source

public class Solution {    public int minPathSum(int[][] grid) {    if(grid.length == 0) return 0;        int[][] dp = new int[grid.length][grid[0].length];                dp[0][0] = grid[0][0];        for(int i = 1; i < grid.length; i++){        dp[i][0] = dp[i - 1][0] + grid[i][0];        }        for(int j = 1; j < grid[0].length; j++){        dp[0][j] = dp[0][j - 1] + grid[0][j];        }                for(int i = 1; i < grid.length; i++){        for(int j = 1; j < grid[0].length; j++){        dp[i][j] = grid[i][j] + Math.min(dp[i - 1][j], dp[i][j - 1]);        }        }        return dp[grid.length - 1][grid[0].length - 1];    }}


Test

    public static void main(String[] args){    int[][] grid = {{1,3,1},{1,5,1},{4,2,1}};    System.out.println(new Solution().minPathSum(grid));    }


0 0
原创粉丝点击