【編程題目 | 100分】k 對(duì)元素最小值 [ 100 / 中等 ]

k 對(duì)元素最小值

題目描述：

• 給定兩個(gè)整數(shù)數(shù)組array1 array2，數(shù)組元素按升序排列假設(shè)從arr1 arr2中分別取出一個(gè)元素，可構(gòu)成一對(duì)元素。
• 現(xiàn)在需要取出k對(duì)元素，并對(duì)取出的所有元素求和，計(jì)算和的最小值。
• 注意：兩對(duì)元素對(duì)應(yīng)arr1 arr2的下標(biāo)是相同的視為同一對(duì)元素

輸入描述:

• 輸入兩行數(shù)組arr1 arr2
• 每行首個(gè)數(shù)字為數(shù)組大小size 0<size<=100
• arr1，2中的每個(gè)元素 0<arr[i]<1000
• 接下來(lái)一行 正整數(shù)k 0<k<=arr1.size * arr2.size

輸出描述：

滿足要求的最小值

示例 1：

輸入

3 1 1 2 3 1 2 3 2

輸出

4

說(shuō)明

• 用例中需要取兩個(gè)元素，
• 取第一個(gè)數(shù)組第0個(gè)元素與第二個(gè)數(shù)組第0個(gè)元素組成一個(gè)元素[1,1]
• 取第一個(gè)數(shù)組第1個(gè)元素與第二個(gè)數(shù)組第0個(gè)元素組成一個(gè)元素[1,1]
• 求和為1+1+1+1=4 ，滿足要求最小

參考代碼

Java代碼實(shí)現(xiàn)：

import java.util.PriorityQueue; import java.util.Scanner;public class minValueOfK {public static void main(String[] args) {Scanner in = new Scanner(System.in);String[] str1 = in.nextLine().split(" ");int m = Integer.parseInt(str1[0]);int[] nums1 = new int[m];for (int i = 1; i < str1.length; i++) {nums1[i - 1] = Integer.parseInt(str1[i]);}String[] str2 = in.nextLine().split(" ");int n = Integer.parseInt(str2[0]);int[] nums2 = new int[n];for (int i = 1; i < str2.length; i++) {nums2[i - 1] = Integer.parseInt(str2[i]);}int k = in.nextInt();// 優(yōu)先隊(duì)列實(shí)現(xiàn),小頂堆int min = 0;PriorityQueue<int[]> queue = new PriorityQueue<>((a, b) -> (nums1[a[0]] + nums2[a[1]]) - (nums1[b[0]] + nums2[b[1]]));// 把nums1索引加入隊(duì)列for (int i = 0; i < Math.min(m, k); i++) {queue.add(new int[]{i, 0});}while (k > 0 && !queue.isEmpty()) {int[] idx = queue.poll();min += (nums1[idx[0]] + nums2[idx[1]]);// 為了避免重復(fù)，把nums2的索引增加if (idx[1] + 1 < n) {queue.add(new int[]{idx[0], idx[1] + 1});}k--;}System.out.println(min);} }

總結(jié)

以上是生活随笔為你收集整理的华为机试：k 对元素最小值的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。