整数对最小和
给定两个整数数组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,为满足要求的最小和
public static void answer() throws IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
String s = bf.readLine();
String[] strArr1=s.split(" ");
s=bf.readLine();
String[] strArr2=s.split(" ");
s=bf.readLine();
int k=Integer.parseInt(s);
int len1=Integer.parseInt(strArr1[0]);
int len2=Integer.parseInt(strArr2[0]);
int[] arr1= new int[len1];
int[] arr2= new int[len2];
for (int i = 0; i < len1; i++) {
arr1[i]=Integer.parseInt(strArr1[i+1]);
}
for (int i = 0; i < len2; i++) {
arr2[i]=Integer.parseInt(strArr2[i+1]);
}
//堆里数组格式{sun,index1,index2}
PriorityQueue<int[]> pq=new PriorityQueue<>((a,b)->a[0]-b[0]);//维护一个最小堆,按和升序排列
for (int i = 0; i < len1; i++) {
pq.offer(new int[]{arr1[i]+arr2[0],i,0});//用arr1的每一个和arr2[0]相加初始化堆
}
int resSum=0;
while (k>0&&!pq.isEmpty()){
int[] current=pq.poll();
int sum=current[0];
int index1=current[1];
int index2=current[2];
resSum+=sum;
k--;
if(index2+1<len2){//如果还有元素没加入堆中
pq.offer(new int[]{arr1[index1]+arr2[index2+1],index1,index2+1});//看是谁的下一个该入队了
}
}
System.out.println(resSum);
// int[] res=new int[len1*len2];
// for (int i = 0; i < len1; i++) {
// for (int j = 0; j < len2; j++) {
// res[i*len2+j]=arr1[i]+arr2[j];
// }
// }
// Arrays.sort(res);
// int count=0;
// for (int i = 0; i < k; i++) {
// count+=res[i];
// }
// System.out.println(count);
}
注意:两对元素如果对应于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,为满足要求的最小和
public static void answer() throws IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
String s = bf.readLine();
String[] strArr1=s.split(" ");
s=bf.readLine();
String[] strArr2=s.split(" ");
s=bf.readLine();
int k=Integer.parseInt(s);
int len1=Integer.parseInt(strArr1[0]);
int len2=Integer.parseInt(strArr2[0]);
int[] arr1= new int[len1];
int[] arr2= new int[len2];
for (int i = 0; i < len1; i++) {
arr1[i]=Integer.parseInt(strArr1[i+1]);
}
for (int i = 0; i < len2; i++) {
arr2[i]=Integer.parseInt(strArr2[i+1]);
}
//堆里数组格式{sun,index1,index2}
PriorityQueue<int[]> pq=new PriorityQueue<>((a,b)->a[0]-b[0]);//维护一个最小堆,按和升序排列
for (int i = 0; i < len1; i++) {
pq.offer(new int[]{arr1[i]+arr2[0],i,0});//用arr1的每一个和arr2[0]相加初始化堆
}
int resSum=0;
while (k>0&&!pq.isEmpty()){
int[] current=pq.poll();
int sum=current[0];
int index1=current[1];
int index2=current[2];
resSum+=sum;
k--;
if(index2+1<len2){//如果还有元素没加入堆中
pq.offer(new int[]{arr1[index1]+arr2[index2+1],index1,index2+1});//看是谁的下一个该入队了
}
}
System.out.println(resSum);
// int[] res=new int[len1*len2];
// for (int i = 0; i < len1; i++) {
// for (int j = 0; j < len2; j++) {
// res[i*len2+j]=arr1[i]+arr2[j];
// }
// }
// Arrays.sort(res);
// int count=0;
// for (int i = 0; i < k; i++) {
// count+=res[i];
// }
// System.out.println(count);
}
全部评论
相关推荐
点赞 评论 收藏
分享
01-20 11:00
广东实验中学 算法工程师 点赞 评论 收藏
分享
2025-12-14 11:43
黑龙江大学 Java
用微笑面对困难:确实比较烂,可以这么修改:加上大学的qs排名,然后大学简介要写一些,然后硕士大学加大加粗,科研经历第一句话都写上在复旦大学时,主要负责xxxx,简历左上角把学校logo写上,建议用复旦大学的简历模板 点赞 评论 收藏
分享
查看1道真题和解析