Bootstrap

给定两个整数数组,arr1、arr2,数组元素按升序排列

题目描述:


给定两个整数数组,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

输出描述
满足要求的最小值

示例一
输入

3 1 1 2
3 1 2 3
2

输出

4

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

解题思路:

2个数组分别取出元素并且求和放到一个排序的集合中

然后取前k个和

public static void main(String[] args) {
 Scanner sc = new Scanner(System.in);
        String[] str1 = sc.nextLine().split(" ");
        String[] str2 = sc.nextLine().split(" ");
        int count = Integer.parseInt(sc.nextLine());
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < str1.length; i++) {
            for (int j = 0; j < str2.length; j++) {
                list.add(Integer.parseInt(str1[i]) + Integer.parseInt(str2[j]));
            }
        }
        Collections.sort(list);
        int res = 0;
        for (int i = 0; i < count; i++) {
            res += list.get(i);
        }
        System.out.println(res);
}

;