蓝桥杯算法提高题——打包

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　Lazy有N个礼物需要打成M个包裹，邮寄给M个人，这些礼物虽然很便宜，但是很重。Lazy希望每个人得到的礼物的编号都是连续的。为了避免支付高昂的超重费，他还希望让包裹的最大重量最小。
输入格式
　　一行两个整数N和M。
　　一行N个整数，表示N个礼物的重量。
输出格式
　　一个整数，表示最小的最大重量。
样例输入
3 2
1 1 2
样例输出
2
数据规模和约定
　　N, M <= 100,000
　　重量 <= 1,000
代码如下：

def judge(x):
    count = 1
    i = 0
    sum = 0
    while i < len(weight):
        sum += weight[i]
        if count > m:
            return False
        if sum > x:
            sum = 0
            count += 1
            i -= 1
        elif sum == x:
            sum = 0
            count += 1
        i += 1
    return True

n = int(input())
m = int(input())
weight = list(map(int,input().split()))
l = max(weight)
r = sum(weight)
while l < r:
    mid = (l+r)//2
    if judge(mid):
        r = mid
        ans = mid
    else:
        l = mid+1
if judge(ans - 1):
    print(ans - 1)
else:
    print(ans)

此题和我写过的搬走要石道理差不多
可以见链接：蓝桥杯算法题之搬走要石！！！
有什么疑问下方留言！