蓝桥杯 分巧克力

分巧克力

问题描述

儿童节那天有K位小朋友到小明家做客。小明拿出了珍藏的巧克力招待小朋友们。小明一共有N块巧克力，其中第i块是Hi x Wi的方格组成的长方形。
　　为了公平起见，小明需要从这 N 块巧克力中切出K块巧克力分给小朋友们。切出的巧克力需要满足：
　　1. 形状是正方形，边长是整数
　　2. 大小相同
例如一块6x5的巧克力可以切出6块2x2的巧克力或者2块3x3的巧克力。
　　当然小朋友们都希望得到的巧克力尽可能大，你能帮小Hi计算出最大的边长是多少么？输入格式　　第一行包含两个整数N和K。(1 <= N, K <= 100000)
　　以下N行每行包含两个整数Hi和Wi。(1 <= Hi, Wi <= 100000)
　　输入保证每位小朋友至少能获得一块1x1的巧克力。输出格式　　输出切出的正方形巧克力最大可能的边长。样例输入：
2 10
6 5
5 6
样例输出2
数据规模和约定　　峰值内存消耗（含虚拟机） < 256M
　　CPU消耗 < 1000ms

Java实现

系统神坑，明明电脑上都跑出来了，官网上却告诉我超时！估计是优先队列那超时了

import java.util.PriorityQueue;
import java.util.Queue;
import java.util.Scanner;

public class Main {
    public static void main(String[] args){
        Scanner in=new Scanner(System.in);
        int N=in.nextInt();
        int K=in.nextInt();
        int[][] shape=new int[N][2];
        Queue<Integer> queue=new PriorityQueue<>();
        long area=0;
        int queuemin=0;
        for(int i=0;i<N;i++){
            shape[i][0]=in.nextInt();
            shape[i][1]=in.nextInt();
            area+=shape[i][0]*shape[i][1];
            if(queue.size()<K){
                queue.add(Math.min(shape[i][0],shape[i][1]));
            }else{
                if(queuemin==0){
                    queuemin=queue.peek();
                }
                int num=Math.min(shape[i][0],shape[i][1]);
                if(num>queuemin){
                    queue.poll();
                    queue.add(num);
                    queuemin=queue.peek();
                }
            }
        }
        long maxArea=area/K;
        //切好后巧克力块的边长的上界
        int maxSide=(int)Math.pow(maxArea,0.5);
//        System.out.println(queue.peek());
        if(N<K) {
        //原始巧克力块的数量小于需要的巧克力块的数量
            System.out.println(getMaxSide(shape, K, maxSide));
        }else 
        //queue.poll()为/切好后巧克力块的边长的下界 测试用例8，9对应的情况
            System.out.println(getMaxSide(shape, K, maxSide,queue.poll()));

    }

    public static int getMaxSide(int[][] shape,int K,int max){
        int maxSide=1;
        for(int i=max;i>1;i--){
            int CNum=0;
            for(int j=0;j<shape.length;j++){
                CNum+=(shape[j][0]/i)*(shape[j][1]/i);
            }
            if(CNum>=K){
                maxSide=i;
                break;
            }
        }
        return maxSide;
    }
    public static int getMaxSide(int[][] shape,int K,int max,int min){
        int maxSide=min;
        for(int i=min+1;i<=max;i++){
            int CNum=0;
            for(int j=0;j<shape.length;j++){
                CNum+=(shape[j][0]/i)*(shape[j][1]/i);
                if(CNum>=K){
                    maxSide=i;
                    break;
                }
            }
            if(CNum<K) break;
        }
        return maxSide;
    }

}