程序员代码面试指南刷题--第四章.矩阵的最小路径和

题目描述
给定一个 n * m 的矩阵 a，从左上角开始每次只能向右或者向下走，最后到达右下角的位置，路径上所有的数字累加起来就是路径和，输出所有的路径中最小的路径和。
输入描述:

第一行输入两个整数 n 和 m，表示矩阵的大小。

接下来 n 行每行 m 个整数表示矩阵。

输出描述:

输出一个整数表示答案。

示例1

输入

4 4
1 3 5 9
8 1 3 4
5 0 6 1
8 8 4 0

输出

12

解法一：递归

好像时间复杂度太高了。。。

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] ss = br.readLine().trim().split(" ");
        int row = Integer.parseInt(ss[0]);
        int col = Integer.parseInt(ss[1]);
        //建立数组
        int[][] arr = new int[row][col];
        for(int i=0;i<row;i++){
            String[] tmp = br.readLine().trim().split(" ");
            for(int j=0;j<col;j++){
                arr[i][j] = Integer.parseInt(tmp[j]);
            }
        }
        int res = findMin(arr,row-1,col-1);
        System.out.println(res);
    }
    public static int findMin(int[][] arr,int i,int j){
        if(i<0||j<0) return Integer.MAX_VALUE;
        if(i==0&&j==0) return arr[i][j];
        int up = findMin(arr,i-1,j);
        int left = findMin(arr,i,j-1);
        return up>left?left+arr[i][j]:up+arr[i][j];
    }
}

解法二：动态规划

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] ss = br.readLine().trim().split(" ");
        int row = Integer.parseInt(ss[0]);
        int col = Integer.parseInt(ss[1]);
        //建立数组
        int[][] arr = new int[row][col];
        for(int i=0;i<row;i++){
            String[] tmp = br.readLine().trim().split(" ");
            for(int j=0;j<col;j++){
                arr[i][j] = Integer.parseInt(tmp[j]);
            }
        }
        int res = findMin(arr,row,col);
        System.out.println(res);
    }
    public static int findMin(int[][] arr,int row,int col){
        if(arr==null||arr[0]==null||arr.length==0||arr[0].length==0) return 0;
        if(row==1&&col==0) return arr[0][0];
        int[][] dp = new int[row][col];
        dp[0][0] = arr[0][0];
        for(int i=1;i<row;i++){
            dp[i][0] = dp[i-1][0]+arr[i][0];
        }
        for(int j=1;j<col;j++){
            dp[0][j] = dp[0][j-1]+arr[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])+arr[i][j];
            }
        }
        return dp[row-1][col-1];
    }
}

解法三：动态规划（压缩空间）

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] ss = br.readLine().trim().split(" ");
        int row = Integer.parseInt(ss[0]);
        int col = Integer.parseInt(ss[1]);
        //建立数组
        int[][] arr = new int[row][col];
        for(int i=0;i<row;i++){
            String[] tmp = br.readLine().trim().split(" ");
            for(int j=0;j<col;j++){
                arr[i][j] = Integer.parseInt(tmp[j]);
            }
        }
        int res = findMin(arr,row,col);
        System.out.println(res);
    }
    public static int findMin(int[][] arr,int row,int col){
        if(arr==null||arr[0]==null||arr.length==0||arr[0].length==0) return 0;
        if(row==1&&col==0) return arr[0][0];
        int more = Math.max(row,col);
        int less = Math.min(row,col);
        boolean morerow = more==row;
        int[] dp = new int[less];
        dp[0] = arr[0][0];
        for(int i=1;i<less;i++){
            dp[i] = dp[i-1]+(morerow?arr[0][i]:arr[i][0]);
        }
        for(int i=1;i<more;i++){
            dp[0] = dp[0]+(morerow?arr[i][0]:arr[0][i]);
            for(int j=1;j<less;j++){
                dp[j] = Math.min(dp[j-1],dp[j])+(morerow?arr[i][j]:arr[j][i]);
            }
        }
        return dp[less-1];
    }
}