题目描述:
给定两个整数数组 array1 array2
数组元素按升序排列
假设从array1 array2中分别取出一个元素可构成一对元素
现在需要取出K个元素
并对取出的所有元素求和
计算和的最小值
注意:两对元素如果对应于array1 array2中的两个下标均相同,则视为同一个元素
输入描述:
输入两行数组array1 array2
每行首个数字为数组大小 size( 0 < size <= 100)
0 < array1(i) <= 10000 < array2(i) <= 1000
接下来一行为正整数k (0 < k <= array1.size() * array2.size())
输出描述:
满足要求的最小和
示例1
输入:
3 1 1 2
3 1 2 3
2
输出:
4
说明:
用例中,需要取两个元素 取第一个数组第0个元素 与第二个数组第0个元素,组成一对元素[1,1]
取第一个数组第1个元素 与第二个数组第0个元素,组成一对元素[1,1]
求和为1+1+1+1=4 为满足要求的最小和
JS参考解题:
解题思路:
可以考虑使用优先队列(最小堆)来实现。首先,定义一个元素为一个包含两个数组中元素之和以及这两个元素在各自数组中索引的对象。然后,将第一个数组的每个元素与第二个数组的第一个元素相加,并将这些组合放入优先队列中。接下来,每次从队列中取出和最小的一对元素,然后将该元素对应在第二个数组中的下一个元素与第一个数组中的该元素相加,形成新的一对元素放回队列,直到取出K对元素为止。
function minSum(array1, array2, k) {
// 解析输入数组
array1 = array1.slice(1); // 移除数组大小,直接获取数组元素
array2 = array2.slice(1);
// 优先队列,这里使用数组实现,实际使用时可以选择性能更优的数据结构
const minHeap = [];
const push = (arr) => minHeap.push(arr) && minHeap.sort((a, b) => a[0] - b[0]);
const pop = () => minHeap.shift();
// 初始化优先队列
for (let i = 0; i < array1.length && i < k; i++) {
push([array1[i] + array2[0], i, 0]); // 数组中保存[当前和, array1中的索引, array2中的索引]
}
let sum = 0;
while (k > 0) {
const [currentSum, i, j] = pop(); // 取出当前和最小的一对
sum += currentSum; // 累加到总和中
if (j + 1 < array2.length) { // 如果array2中还有元素,将下一个元素加入优先队列
push([array1[i] + array2[j + 1], i, j + 1]);
}
k--;
}
return sum;
}
// 示例输入
const array1 = [3, 1, 1, 2];
const array2 = [3, 1, 2, 3];
const k = 2;
// 调用函数并输出结果
console.log(minSum(array1, array2, k));