Friday, February 7, 2014

Triangle (Java)


LeetCode

Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.
For 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).
Note:
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.
Solution: DP (met during Amazon interview)

Declare pathSum array with length of triangle size(). For triangle, the bottom row length is equal to the height of triangle, so use pathSum to hold the bottom row's value, then from bottom to up, find minimum path
/*
Given a triangle, find the minimum path sum from top to bottom. Each step you may move to adjacent numbers on the row below.
For 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).
Note:
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.
*/
public class Solution {
public int minimumTotal(ArrayList<ArrayList<Integer>> triangle) {
if (triangle==null|| triangle.size()==0){
return 0;
}
// declare an int[] pathSum with length of triangle.size(), witch used to record the current path sum from bottom to up
int[] pathSum=new int[triangle.size()];
// calculate minimum path sum from bottom to up
int rowNum=triangle.size();
for (int row=rowNum-1; row>=0; row--){
int colNum=triangle.get(row).size();
for (int col=0; col<colNum; col++){
if (row==rowNum-1){
// from bottom to up, current is bottom level
pathSum[col]=triangle.get(row).get(col);
}
else{
// if not bottom level, so from previous level wich are store in pathSum find smaller value can access current point,
//then update it
pathSum[col]=Math.min(pathSum[col], pathSum[col+1])+triangle.get(row).get(col);
}
}
}
// right no, the pathSum[0] contains the minimum path sum
return pathSum[0];
}
}
view raw Triangle.java hosted with ❤ by GitHub

No comments:

Post a Comment