华为机试：k 对元素最小值

package org.example;

import javafx.util.Pair;

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

/**
 * k 对元素最小值
 *
 * @author zey
 * @since 2022-06-12
 */
public class MinPairOfK {

    /*
    题目描述：
    给定两个整数数组array1 array2，数组元素按升序排列假设从arr1 arr2中分别取出一个元素，可构成一对元素。
    现在需要取出k对元素，并对取出的所有元素求和，计算和的最小值。
    注意：两对元素对应arr1 arr2的下标是相同的视为同一对元素
    输入描述:
    输入两行数组arr1 arr2
    每行首个数字为数组大小size 0<size<=100
    arr1，2中的每个元素 0<arr[i]<1000
    接下来一行 正整数k 0<k<=arr1.size * arr2.size
    输出描述：
    满足要求的最小值

    示例 1：
    输入
    3 1 1 2
    3 1 2 3
    2
    输出
    4
    说明
    用例中需要取两个元素，
    取第一个数组第0个元素与第二个数组第0个元素组成一个元素[1,1]
    取第一个数组第1个元素与第二个数组第0个元素组成一个元素[1,1]
    求和为1+1+1+1=4 ，满足要求最小
    */
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        String[] arr1 = input.nextLine().split(" ");
        String[] arr2 = input.nextLine().split(" ");
        int count = input.nextInt();

        List<Integer> arr1List = transformArrToList(arr1);
        List<Integer> arr2List = transformArrToList(arr2);
        List<Pair<Integer, Integer>> pairs = new ArrayList<>();
        arr1List.forEach(a1 -> {
            arr2List.forEach(a2 -> {
                pairs.add(new Pair<>(a1, a2));
            });
        });

        List<Pair<Integer, Integer>> sortList = pairs.stream().sorted((o1, o2) -> o1.getKey() + o1.getValue() - o2.getKey() - o2.getValue()).collect(Collectors.toList());
        int sum = 0;
        for (int i = 0; i < count; i++) {
            Pair<Integer, Integer> pair = sortList.get(i);
            sum += pair.getKey() + pair.getValue();
        }
        System.err.println(sum);
    }
    
    private static List<Integer> transformArrToList(String[] arr1) {
        List<Integer> arr1List = new ArrayList<>();
        for (int i = 1; i < arr1.length; i++) {
            arr1List.add(Integer.valueOf(arr1[i]));
        }
        return arr1List;
    }
}