蓝桥杯 算法训练 algo_1003 如何用python实现

问题描述
  JiaoShou在爱琳大陆的旅行完毕,即将回家,为了纪念这次旅行,他决定带回一些礼物给好朋友。
  在走出了怪物森林以后,JiaoShou看到了排成一排的N个石子。
  这些石子很漂亮,JiaoShou决定以此为礼物。
  但是这N个石子被施加了一种特殊的魔法。
  如果要取走石子,必须按照以下的规则去取。
  每次必须取连续的2*K个石子,并且满足前K个石子的重量和小于等于S,后K个石子的重量和小于等于S。
  由于时间紧迫,Jiaoshou只能取一次。
  现在JiaoShou找到了聪明的你,问他最多可以带走多少个石子。
输入格式
  第一行两个整数N、S。
  第二行N个整数,用空格隔开,表示每个石子的重量。
输出格式
  第一行输出一个数表示JiaoShou最多能取走多少个石子。
样列输入
  8 3
  1 1 1 1 1 1 1 1
样列输出
  6
样列解释
  任意选择连续的6个1即可。
数据规模和约定
  对于20%的数据:N<=1000
  对于70%的数据:N<=100,000
  对于100%的数据:N<=1000,000,S<=10^12,每个石子的重量小于等于10^9,且非负

在网上看到用的是二分法,我自己也就尝试用了二分法。改了好多次还是有80%测试点超时。
我自己的代码:

n,s=map(int,input().split())
group=list(map(int,input().split()))
a=len(group)+1
left,r = 0,((a-1)//2)   #left,r为最左最右的序列
val=[0]*(a)
for i in range(1,a):
    val[i]=val[i-1]+group[i-1]   #求前i项的和,觉得sum()函数费时
def check(tem):     #判断是否满足条件
    for i in range(tem,n-tem+1):
        if (val[i]-val[i-tem])<=s and (val[i+tem]-val[i])<=s: 
            return True
    return False

while left<r:        #二分法
    mid=(left+r)//2
    nums=mid+1
    if check(nums): 
        left=mid+1
    else:
        r=mid 
print(2*left)
'''
恳请指点,将代码的时间复杂度变得更低

将第5行至第7行改成列表表达式试试:

val=[sum(group[0:i]) for i in range(a)]