蓝桥杯每日一练——礼物 python

最新推荐文章于 2023-04-06 16:21:06 发布

原创最新推荐文章于 2023-04-06 16:21:06 发布 · 1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #蓝桥杯

这篇博客讨论了一个关于旅行者JiaoShou在爱琳大陆遇到的魔法石子问题。问题要求每次取走连续的2*K个石子，其中前K个和后K个石子的重量之和都必须小于等于S。JiaoShou只能取一次。博主通过二分法尝试解决，但遇到了超时问题。文章中提供了二分查找的Python代码实现，但仅适用于20%的数据。博客探讨了调整二分法策略的可能性以应对更大规模的数据。

问题描述
　　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，且非负

小的不才，只能做到按题目要求输出正确答案，但是只能获得20分，后面就运行超时了

一开始说到二分法，总是从题目中的1 1 1 1 1 1 1 1里面去分，但是很明显这样分做不了题目
换个思路，二分法可以从k去分，错了就缩小mid

感觉二分法基本都是从想要获得的答案去分

def fun(mid):
    a = 0
    sum = 0
    # 先算第一个k，才能确定第二个k的位置
    for i in range(n-mid):               # n-mid 再后面都不够给第二个k了，没必要再算下去
        sum += lists[i]
        a += 1
        if a == mid and sum <= s:        # 满足条件 那就看看第二个k行不行
            sum1 = 0
            for j in range(mid):
                sum1 += lists[j+i+1]
            if sum1 <= s:                # 第二个k也行的话 那mid就是可以完成任务的k
                return True
            else:
                sum -= lists[i - a + 1]  # 第二个不行的话 记得把第一个退掉第一个，再进入下一个循环 否则就会出现错误
                a -= 1
        if sum > s:
            sum -= lists[i-a+1]
            a -= 1
    return False


n,s = map(int,input().split())
lists = list(map(int,input().split()))
l = 1                              # 左
r = n // 2                         # 右
answer = 0
# 接下来就是二分法的常规操作
while r >= l:
    mid = (r+l)//2
    if fun(mid):
        l = mid+1
        answer = mid
    else:
        r = mid-1
if fun(mid+1):
    answer += 1
print(answer*2)