阿里 3月25日 二维矩阵列选元素最小和

给一个长度为n,宽度为3的3*n矩阵,在其每一列取一个数组成一个数组,找到这种数组从i=1到n-1的右一元素减左一元素的绝对值的最小和.

import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;
public class Test {
    static HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int[][] a = new int[3][n];
        for (int i = 0; i < 3; i++) {
            for (int j = 0; j < n; j++) {
                a[i][j] = in.nextInt();
            }
        }
        int sum = 0;
        
        // 动规
        int[][] res = minPathSum(a, n);
        sum = res[0][n-1];
        for(int i=1;i < 3; i++){
            if(sum>res[i][n-1]){
                sum = res[i][n-1];
            }
        }

        // 以下为不用动规的方法,其实是模拟动规,每两列之间选择最小,贪心.
        //优点是找到了每一个索引,可以生成那个数组
        // int tmp = minIndex(a,0);
        // sum += tmp;
        
        // for (int j = 1; j < n - 2; j++) {
        //     tmp = minDiff(a,j,map.get(tmp));
        //     sum += tmp;
        // }
        System.out.println(sum);
    }

    public static int minDiff(int[][] nums1, int j ,int i) {
        int[] a = new int[3];
        int index =0;
        map.clear();
        for (int k = 0; k < 3; k++){
            a[index++] = Math.abs(nums1[i][j] - nums1[k][j + 1]);
            map.put(a[index-1],k);
        }
        Arrays.sort(a);
        return a[0];
    }

    public static int minIndex(int[][] nums1, int j) {
        int[] a = new int[9];
        int index = 0;
        for (int i = 0; i < 3; i++) {
            for (int k = 0; k < 3; k++) {
                a[index++] = Math.abs(nums1[i][j] - nums1[k][j + 1]);
                map.put(a[index-1],k);
            }
        }
        Arrays.sort(a);
        return a[0];
    }
    public static int[][] minPathSum(int[][] a, int n){
        int tmp[]={0,0,0};
        int[][] dp = new int[3][n];
        // 生成邻差绝对值求和的状态转移结果数组dp[][]
        for(int j=1;j<n;j++){
            // i遍历的是当前所在列
            for(int i=0;i<3;i++){
                // k遍历的是上一个列
                for(int k=0;k<3;k++){
                    // 生成每一列每个元素a[i][j]的邻差绝对值的和
                    tmp[k] = dp[k][j-1] + Math.abs(a[i][j] - a[k][j-1]);
                }
                // 取每一列每个元素的邻差绝对值求和的最小值放入dp[][]
                dp[i][j] = min(tmp);
            }
        }
        return dp;
    }
    public static int min(int[] tmp){
        int min = tmp[0];
        for(int num : tmp)
            if(min>num)
                min =num;
        return min;
    }
}