算法实践：月度开销（二分）

月度开销

描述

农夫约翰是一个精明的会计师。他意识到自己可能没有足够的钱来维持农场的运转了。他计算出并记录下了接下来N(1 ≤N≤ 100,000) 天里每天需要的开销。

约翰打算为连续的M(1 ≤M≤N) 个财政周期创建预算案，他把一个财政周期命名为fajo月。每个fajo月包含一天或连续的多天，每天被恰好包含在一个fajo月里。

约翰的目标是合理安排每个fajo月包含的天数，使得开销最多的fajo月的开销尽可能少。

输入

第一行包含两个整数N,M，用单个空格隔开。

接下来N行，每行包含一个1到10000之间的整数，按顺序给出接下来N天里每天的开销。

输出

一个整数，即最大月度开销的最小值。

样例

7 5
100
400
300
100
500
101
400

500

难度

中等，二分

解题思路

月度开销的值应该在 每天开销最大值 和 每天开销的总和 之间，通过二分法不断逼近。judge函数判断按照mid开销所分的周期，如果周期大于给定周期M，则可以增加开销，反之，减小开销。

代码

def judge(mid,l,n,m):
    summary = 0
    t = 0
    for i in range(n):
        if(summary+l[i]>mid):  #如果前几个月加上这个月的钱大于这次尝试的fajo月开销，那么说明这个月的钱应该放在下个fajo月里 
            summary = l[i]
            t += 1
        else:
            summary += l[i]
     #继续增加开销 
    if(t>m):
        return True
    #减小开销
    else:
        return False

x = input().split()
n = (int)(x[0])
m = (int)(x[1])

Max = 0 
total = 0
a = []
for i in range(n):
    temp = (int)(input())
    a.append(temp)
    total += temp  #记录开销的总和 
    if(Max<temp):
        Max = temp   #记录开销的最大值 
l = Max
r = total
while(l<=r):
    mid = (int)((l+r)/2)
    if(judge(mid,a,n,m)):
        l = mid+1
    else:
        r = mid-1
print(mid)