问题描述
住在有顶天的天人Tensi对自己的住处很不满。终于有一天她决定把门前碍眼的要石通通丢掉(怒扔要石)。控制要石自然是很容易的事,不过也会消耗灵力。假设搬走一块质量为1的要石会消耗1点灵力,而且由于要石都是连着放置的缘故所以每次除了搬走一颗,也可以搬走连续的任意数量的要石,自然质量是算在一起的。现在Tensi准备最多使用M次灵力,但是她太懒……所以每次只会使用同量的灵力,也因为她太烂,所以也不愿意多花一点灵力……现在很懒的Tensi需要你帮她计算最少一次需要消耗多少灵力,能够在M次内把所有要石都丢到人间去……
输入格式
第一行两个数N,M,用一个空格分隔。1<=n<=1000,1<=m<=400
表示一共有N颗要石需要搬走已经Tensi最多发动M次灵力。 接下来包括N 个正整数 0<=ai<=40000
顺序表示每一颗要石的质量。
输出格式
输出一个数T
表示Tensi 每次至少消耗T灵力。0<=T<=1000000
如果无解输出-1.
样例输入
5 3
1 2 1 1 1
样例输出
3
数据规模和约定
对于100%的数据,1<=n<=1000,1<=m<=400,0<=ai<=40000。
保证0<=T<=1000000。
def fun(x):
count = 1
sum = 0
i = 0
while i < len(stone): # 用i循环来捡石头
sum += stone[i] # 捡起一块石头
# 如果这是最后一块石头,并且捡的起来(力气够),那就返回真
if i == len(stone) - 1 and sum <= x:
return True
if sum > x: # 如果sum > x,也就是力气不够
sum = 0 # 那就只能把前面的先扔下去了
count += 1 # 次数+1
i -= 1 # 既然力气不够,第i块只能重新捡起来,i-1就是把i循环退回到捡不起来的那颗
elif sum == x: # 如果力气刚好够
sum = 0 # 那就把手上的都扔下去
count += 1 # 次数+1
if count > m: # 次数超了
return False # 返回假
# if count == m and i != len(stone) - 1:
# return False
i += 1 # i循环日常 +1
n, m = map(int, input().split())
stone = list(map(int, input().split()))
l = 0 # 左
# 题意可知,答案 不能小于 所有石头重量/次数,再小就完不成任务
for i in stone:
l += i
l //= m
# 二分法的精髓就是 左和右不断缩小,直到相同
r = 1000000 # 右
min = 0 # 答案
# 开始!
while l < r:
mid = (l + r) // 2 # 二分法常规操作 取中间数
if fun(mid): #判断是否能搬完石头
r = mid - 1 # 可以的话,右缩小
min = mid # 把这个可以的值赋值给min
else:
l = mid + 1 # 不行就左边缩小咯
if min == 0: # 根据题意,min=0 就是没有解,输出-1
print(-1)
# 这个算法很奇怪(应该是我写的有问题),右的值不一样会出现计算出来的结果不一样,
#有时候会比答案高1,所以用这个蠢办法,测试-1成不成立
elif fun(min - 1):
print(min - 1)
else:
print(min)
看完代码,可能你会问我,为什么不用for循环,因为用不了啊,for循环是不能在循环里边改变i的值的,就算变了,也只是这次循环变了,下次循环还是那个数,所以这里只能用while循环
这道题会有一个误导,每次只会使用同量的灵力,会让考生误以为每一次都必须使用这么多的灵力,少一点都不行,但其实只是不能多于这个数而已,
比如:
5 3
16 14 13 15 10
答案:27
分成三次拿取:16 (14+13) (15+10)
三次都不会高于27,所以答案为27
下面是我一开始以为必须每次使用相同的灵力写的代码:
count = 0
for i in range(len(stone)):
count += stone[i]
if count == x:
count = 0
if count > x:
return False
if i == len(stone) - 1 and count != 0:
return False
if i == len(stone) - 1 and count == 0:
return True
n,m = map(int,input().split())
stone = list(map(int,input().split()))
l = 0
for i in stone:
l += i
l //= m
while l < 10000:
if fun(l):
print(l)
break
l += 1