[JAVA]动态规划-求三角矩阵的最佳路径

分别从三角形顶部和底部开始

import java.util.*;

/**
 * Created by dongua on 2016/10/20.
 */
public class Main {

    static int totalRow;
    static ArrayList<ArrayList<Integer>> triangle = new ArrayList<>();

    public static void main(String[] args) {

        ArrayList<Integer> a = new ArrayList<>();
        a.add(2);
        ArrayList<Integer> b = new ArrayList<>();
        b.add(3);
        b.add(4);
        ArrayList<Integer> c = new ArrayList<>();
        c.add(6);
        c.add(5);
        c.add(7);
        ArrayList<Integer> d = new ArrayList<>();
        d.add(4);
        d.add(1);
        d.add(8);
        d.add(3);
        triangle.add(a);
        triangle.add(b);
        triangle.add(c);
        triangle.add(d);
        totalRow = 4;

        int res1 =minimumTotalFromBottom(triangle);
        int res2 = minimumTotalFromTop(triangle);

        System.out.println(res1 +" "+res2);
    }

public static int minimumTotalFromBottom(ArrayList<ArrayList<Integer>> triangle) {
        int len = triangle.size();
        int[][] arr = new int[len][len];

        ArrayList<Integer> lastRow = triangle.get(len-1);
        for(int i=0;i<lastRow.size();i++){
            arr[len-1][i]=lastRow.get(i);//填充最后一行
        }
        for(int i=len-2;i>=0;i--){
            lastRow = triangle.get(i);
            for(int j=0;j<lastRow.size();j++){
                arr[i][j] = lastRow.get(j) + Math.min(arr[i+1][j],arr[i+1][j+1]);
            }
        }

        return arr[0][0];
    }

    public static int minimumTotalFromTop(ArrayList<ArrayList<Integer>> t) {
        int len = t.size();
        int[][] arr = new int[len][len];
        arr[0][0] = t.get(0).get(0);
        for(int i=1;i<len;i++){
            ArrayList<Integer> tempRow = t.get(i);
            for(int j=0;j<tempRow.size();j++){
                if(j==0){
                    arr[i][j] = tempRow.get(j)+arr[i-1][j];
                }else if(j==tempRow.size()-1){
                    arr[i][j] = tempRow.get(j)+arr[i-1][j-1];
                }else{
                    arr[i][j] = tempRow.get(j)+Math.min(arr[i-1][j],arr[i-1][j-1]);
                }
            }
        }
        int min = Integer.MAX_VALUE;
        for(int i=0;i<triangle.get(len-1).size();i++){
            if(arr[len-1][i]<min){
                min = arr[len-1][i];
            }
        }

        return min;
    }
}