矩陣的最短距離和

ZHUO_SIR發表於2018-07-16

題目: 
給定一個矩陣m,從左上角開始每次只能向右或者向下走,最後到達右下角的為止,路徑上所有的數字累加起來就是路徑和,返回所有路徑中的最小的路徑和。 
舉例: 
1 3 5 9 
8 1 3 4 
5 0 6 1 
8 8 4 0 
路徑1,3,1,0,6,1,0是所有路徑中路徑和最小的,所以返回12. 
這裡寫圖片描述
程式碼:

package com.lyf.dp;

import org.junit.Test;

/**
 * Created by fangjiejie on 2017/5/17.
 */
public class MatrixMinPath {//dp[i][j]代表從起點到達當前位置為止的最短路徑和
    public int getPathLen(int [][]matrix){
        if(matrix==null||matrix.length==0||matrix[0].length==0){
            return 0;
        }
        int row=matrix.length;
        int col=matrix[0].length;
        int [][]dp=new int[row][col];
        dp[0][0]=matrix[0][0];
        for(int i=1;i<row;i++){//找到第一列的dp值
            dp[i][0]=dp[i-1][0]+matrix[i][0];
        }
        for(int j=1;j<col;j++){//找到第一行的dp值
            dp[0][j]=dp[0][j-1]+matrix[0][j];
        }
        for(int i=1;i<row;i++){
            for(int j=1;j<col;j++){
                dp[i][j]=Math.min(dp[i-1][j],dp[i][j-1])+matrix[i][j];//當前dp值為 從左邊、上邊兩個位置取一個最小值,加上本身
            }
        }
        System.out.println("dp矩陣:");
        for(int[] i:dp){
            for(int j:i){
                System.out.print(j+",");
            }
            System.out.println();
        }
        return dp[row-1][col-1];
    }
    @Test
    public void test(){
        int arr[][]={
                {1,3,5,9},{8,1,3,4},{5,0,6,1},{8,8,4,0}
        };

        System.out.println("最短路徑"+getPathLen(arr));
    }
}

相關文章