整数对最小和 - 华为OD统一考试-编程知识

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

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();// 读取 kint 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;}// 读取 kint 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;
}