题目二：K对元素最小值

一、题目描述：

给定两个整数数组，arr1、arr2，数组元素按升序排列；
假设从arr1、arr2中分别取出一个元素，可构成一对元素；
现在需要取出k对元素，并对取出的所有元素求和，计算和的最小值；
注意：两对元素对应arr1、arr2的下标是相同的，视为同一对元素。

二、输入描述

输入两行数组arr1 arr2

每行首个数字为数组大小size， 0 < size <= 100

arr1，arr2中的每个元素e， 0< e <1000

接下来一行，正整数k 0 < k <= arr1.size * arr2.size

三、输出描述

满足要求的最小值

四、示例1

输入

1 1 2
1 2 3
2

输出

4

说明：

用例中需要取两个元素，取第一个数组第0个元素与第二个数组第0个元素组成一个元素[1,1];
取第一个数组第1个元素与第二个数组第0个元素组成一个元素[1,1];
求和为1+1+1+1=4 ,满足要求最小

五、Java

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

public class Main0002 {
	public static void main(String[] args) {
		try (Scanner scanner = new Scanner(System.in)) {
			int[] arr1 = parseArray(scanner.nextLine());
			int[] arr2 = parseArray(scanner.nextLine());
			int k = scanner.nextInt();
			solution(arr1, arr2, k);
		}
	}
	
	private static void solution(int[] arr1, int[] arr2, int k) {
		int initialCapacity = arr1.length * arr1.length;
		List<Integer> sums = new ArrayList<>(initialCapacity);
		for (int x : arr1) {
			for (int y : arr2) {
				sums.add(x + y);
			}
		}
		sums.sort(Integer::compareTo);
		int res = 0;
		for (int i = 0; i < k; i++) {
			res += sums.get(i);
		}
		System.out.println(res);
	}
	
	private static int[] parseArray(String line) {
		String[] split = line.split(" ");
		int[] arr = new int[split.length];
		for (int i = 0; i < arr.length; i++) {
			arr[i] = Integer.parseInt(split[i]);
		}
		return arr;
	}
}