阿里 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;
        // 第一個元素有三種選擇,後面元素已經選擇好
        // 因此單獨做了一個minDiff()的姐妹版minIndex()
        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];
    }
}