0
点赞
收藏
分享

微信扫一扫

蓝桥杯算法训练之分礼物(python组)

Ichjns 2022-03-17 阅读 62

资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
  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<=1012,每个石子的重量小于等于109,且非负

解题思路:这道题很有意思,我把它理解为区间的划分。使用二分法:

代码如下:

def quan(mid):
    for i in range(mid,n):
        if i+mid <=n:
            if (sum1[i] - sum1[i - mid] <= s and sum1[i + mid] - sum1[i]<=s):
                return 1
    return 0
sum1 = []
sum2 = 0
n,s = map(int,input().split())
list1  = list(map(int,input().split()))
for i in range(0,n):
    sum2 += list1[i]
    sum1.append(sum2)
sum1.insert(0,0)
l = 1
r = n//2
while(l<=r):
    mid = (l+r) // 2
    if quan(mid):
        l = mid +1
        k  = mid
    else:
        r = mid - 1
print(2*k)

不懂留言

举报

相关推荐

0 条评论