整数对最小和

给定两个整数数组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);
    }
全部评论

相关推荐

2025-12-14 11:43
黑龙江大学 Java
用微笑面对困难:确实比较烂,可以这么修改:加上大学的qs排名,然后大学简介要写一些,然后硕士大学加大加粗,科研经历第一句话都写上在复旦大学时,主要负责xxxx,简历左上角把学校logo写上,建议用复旦大学的简历模板
点赞 评论 收藏
分享
迷茫的大四🐶:不是,匿名发帖,你也可以发
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务