整数对最小和 - 华为OD机试题

2024-07-26 13:39:53 浏览数 (2)

题目描述

给定两个整数数组array1、array2,数组元素按升序排列。

假设从array1、array2中分别取出一个元素可构成一对元素,现在需要取出k对元素,并对取出的所有元素求和,计算和的最小值。

注意:两对元素如果对应于array1、array2中的两个下标均相同,则视为同一对元素。

输入描述

输入两行数组array1、array2,每行首个数字为数组大小size(0 < size <= 100);

0 <array1[i] <=1000

0 <array2[i] <= 1000

接下来一行为正整数 k

0 < k <= array1.size() * array2.size()

输出描述

满足要求的最小和

示例一

代码语言:javascript复制
输入:
3 1 1 2
3 1 2 3
2

输出:
4

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

java题解

题解

代码语言:javascript复制
数据量很小, 直接暴力枚举所有的数对,然后对数对和排序,取前 k 个最小的数对求和即为答案。
代码语言:javascript复制
import java.util.Arrays;
import java.util.Scanner;
/**
 * @author code5bug
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 读取第一个数组
        int[] array1 = Arrays.stream(scanner.nextLine().split(" "))
                .mapToInt(Integer::parseInt).toArray();

        // 读取第二个数组
        int[] array2 = Arrays.stream(scanner.nextLine().split(" "))
                .mapToInt(Integer::parseInt).toArray();

        // 读取 k
        int k = scanner.nextInt();

        // 枚举所有的数对元素和,存入 t 中,并排序
        int[] t = new int[array1[0] * array2[0]];
        int index = 0;
        for (int i = 1; i < array1.length; i  ) {
            for (int j = 1; j < array2.length; j  ) {
                t[index  ] = array1[i]   array2[j];
            }
        }
        Arrays.sort(t);

        // 取前 k 个求和输出结果
        int result = 0;
        for (int i = 0; i < k; i  ) {
            result  = t[i];
        }
        System.out.println(result);
    }
}

0 人点赞