整数对最小和

给定两个整数数组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-05 18:09
已编辑
广东药科大学 后端工程师
点赞 评论 收藏
分享
程序员花海:实习和校招简历正确格式应该是教育背景+实习+项目经历+个人评价 其中项目经历注意要体现业务 实习经历里面的业务更是要自圆其说 简历模板尽可能保持干净整洁 不要太花哨的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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