华为OD机试 Python【求数组组合最小和的k对元素】

题目

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

解题思路

使用小根堆（最小堆）来高效地找到两个升序数组中和最小的k对元素。首先初始化一个最小堆，将所有可能的元素对插入堆中，然后从堆中取出和最小的k对元素并计算它们的总和。通过堆的特性确保每次取出的都是当前最小的元素对，保证了结果的最小性。

代码

import heapq