整数对最小和 - 华为OD统一考试

news2024/11/14 14:32:44

OD统一考试（C卷）

分值： 100分

题解： Java / Python / C++

alt

题目描述

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

输出描述

满足要求的最小和

示例1

输入：
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，为满足要求的最小和

题解

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

Java

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

Python

array1 = list(map(int, input().split()))
array2 = list(map(int, input().split()))
k = int(input())

# 枚举所有的数对元素和，存入 t 中，并排序
t = []
for i in range(1, array1[0] + 1):
    for j in range(1, array2[0] + 1):
        t.append(array1[i] + array2[j])
t.sort()

# 取前 k 个求和输出结果
print(sum(t[:k]))

C++

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    int num;
    // 读取第一个数组
    vector<int> array1;
    while(cin >> num) {
        array1.push_back(num);
        if(cin.peek() == '\n') break;
    }

    // 读取第二个数组
    vector<int> array2;
    while(cin >> num) {
        array2.push_back(num);
        if(cin.peek() == '\n') break;
    }

    // 读取 k
    int k;
    cin >> k;

    // 枚举所有的数对元素和，存入 t 中，并排序
    vector<int> t;
    for (size_t i = 1; i < array1.size(); ++i) {
        for (size_t j = 1; j < array2.size(); ++j) {
            t.push_back(array1[i] + array2[j]);
        }
    }
    sort(t.begin(), t.end());

    // 取前 k 个求和输出结果
    int result = 0;
    for (int i = 0; i < k && i < t.size(); ++i) {
        result += t[i];
    }
    cout << result << endl;

    return 0;
}