Triangle

LintCode-109. Triangle

Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.

Notice

Bonus point if you are able to do this using only O(n) extra space, where n is the total number of rows in the triangle.

Example

Given the following triangle:

[
     [2],
    [3,4],
   [6,5,7],
  [4,1,8,3]
]

The minimum path sum from top to bottom is 11 (i.e., 2 + 3 + 5 + 1 = 11).

public class Solution {
    /**
     * @param triangle: a list of lists of integers.
     * @return: An integer, minimum path sum.
     */
    public int minimumTotal(int[][] triangle) {
        if (triangle == null || triangle.length == 0) {
        	return -1;
        }

        int n = triangle[triangle.length - 1].length;
        int[] dp = new int[n];

        for (int i = 0; i < n; i++) {
        	dp[i] = triangle[triangle.length - 1][i];
        }

        for (int i = n - 2; i >= 0; i--) {
        	for (int j = 0; j < triangle[i].length; j++) {
        		dp[j] = Math.min(dp[j], dp[j + 1]) + triangle[i][j];
        	}
        }

        return dp[0];
    }
}

Hope this helps,
Michael

DigitalOcean Referral Badge