题目描述
儿童节那天有 K 位小朋友到小明家做客。小明拿出了珍藏的巧克力招待小朋友们。
小明一共有 N 块巧克力,其中第 i 块是 Hi * Wi 的方格组成的长方形。为了公平起见,
小明需要从这 N 块巧克力中切出 K 块巧克力分给小朋友们。切出的巧克力需要满足:
1.形状是正方形,边长是整数;
2.大小相同;
例如一块 6x5 的巧克力可以切出 6 块 2x2 的巧克力或者 2 块 3x3 的巧克力。
当然小朋友们都希望得到的巧克力尽可能大,你能帮小明计算出最大的边长是多少么?
输入描述
第一行包含两个整数 N 和 K。
以下 N 行每行包含两个整数 Hi , Wi (1<=Hi , Wi <=105)
输入保证每位小朋友至少能获得一块 1x1 的巧克力。
输出描述
输出切出的正方形巧克力最大可能的边长。
输入输出样例
示例1:
输入:输出:
代码
N, K = map(int, input().split()) # 现巧克力数,小朋友数
chocolate = [] # 开二维数组:存每块巧克力的长,宽
for _ in range(N): # 存入过程
H_W = list(map(int, input().split()))
chocolate.append(H_W)
def find(n): # 判断用当前的 mid 作为边长切割,是否大于小朋友的人数
count = 0 # 记录能分多少块
for c in chocolate: # 遍历每个巧克力的长,宽
count += (c[0] // n) * (c[1] // n) # 计算公式,累加 count
if count >= K: # 满足条件说明边长偏小
return True
else: # 不满足说明边长偏大
return False
L, R = 1, max(map(max, chocolate)) # 创建所查找数组的左右指针,以最长的边长作为右指针
while L <= R: # 二分查找合适的边长:mid
mid = (L + R) // 2
if not find(mid): # 边长偏大,左移右指针范围
R = mid - 1
else: # 边长偏小,右移左指针范围
L = mid + 1
print(R) # 右指针为最后答案
核心思想
想到用二分法找最优解
详解
利用二分查找法去找最合适的边长,最合适是什么意思呢,就是既要够k个小朋友人,又要边长尽可能大。用二分法,最小边长定为左端点为1,最大边长定为右端点,mid就为切割边长,然后通过计算公式算出可以分给几个小朋友。若count<k,意思是不够分,说明边长偏大,还得往小分,此时右端点变为mid-1,这是为什么呢?
对于mid右侧的点,都是比切割边长大的点,都不符合题意,mid本身也不满足条件,所以减1。
补充二分法的check函数思想:
那计算公式又为什么是count += (c[0] // n) * (c[1] // n)呢?
我们这里是两块或者多块蛋糕,所以要累加。