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

小土豆Yuki

发布于 2024-07-26 13:39:53

370

发布于 2024-07-26 13:39:53

文章被收录于专栏：洁癖是一只狗

题目描述

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

输出描述

满足要求的最小和

示例一

输入：
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题解

题解

数据量很小， 直接暴力枚举所有的数对，然后对数对和排序，取前 k 个最小的数对求和即为答案。

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);
    }
}

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2024-07-24，如有侵权请联系 cloudcommunity@tencent.com 删除

排序