P4343 [SHOI2015]自动刷题机 Python(二分答案)

cactusPu

已于 2023-01-03 09:26:17 修改

阅读量171

点赞数

分类专栏： python 文章标签： python

于 2023-01-02 19:26:46 首次发布

本文链接：https://blog.csdn.net/WoBuHuiC/article/details/128523416

版权

python 专栏收录该内容

24 篇文章 3 订阅

订阅专栏

在这吐槽一下洛谷的OJ，对Python不是很友好，一样的思路实现下来大部分数据会TLE，需要不断的优化复杂度：（

题目地址：[SHOI2015]自动刷题机 - 洛谷

优化不下去了，对比了一下PythonAC的大佬，思路大致相同，虽然得分80，但就当AC了吧：）

此题目大体意思：每行一个整数x代表一秒钟输入x行代码，若为负数则代表删除了x行代码(0行删除代表一行代码也没有输入)，切了k题代表AC了k道题。n代表要AC题目所需要的代码行数，求出所需n的最大值与最小值。

由题意得我们可以枚举答案，再一个个去check()，但暴力枚举必然会超时，所以我们可以二分答案来降低时间复杂度，因为需要求一个最大值和一个最小值，所以我们要二分两次求出答案。

check()函数我们用来判断当我们输入n后其切题的次数，再在二分循环判断n是大是小。

先输入数据：

l, k = (int(x) for x in input().split(' '))
OJ = []
min_ans = 0
max_ans = 0
for i in range(l):
    OJ.append(int(input()))

在这我们先求最大值(先求最小值也可以)，先写最大值的二分：

left = 1
right = sum(abs(i) for i in OJ)  # 定义右边界，这里可以定义为int(1e9)，但这会导致部分数据TLE，所以我们先求出数组的绝对值和
while left <= right:  # 求最大值
    mid = (left + right) // 2
    if check(mid) >= k:  # 切题次数太多，说明此答案较小
        left = mid + 1  # 移动右边界
        if check(mid) == k:
            max_ans = mid
    else:  # 切题次数太少，说明此答案较大
        right = mid - 1  # 移动左边界

强调的是为什么 check(mid) >= k:后内部还有一个if判断？

因为当check(mid) >= k时,我们还需要将left移至mid再次循环以取到最大值，当 > k时我们得出的答案是偏小的，即使 == k时也可能还有比k还大的数也能够满足条件。

再求最小值(操作与求最大值差不多，不多说):

left = 1
right = max_ans  # 右边界设为我们刚求的最大值，减少循环次数
while left <= right:  # 求最小值
    mid = (left + right) // 2
    if check(mid) <= k:
        right = mid - 1
        if check(mid) == k:
            min_ans = mid
    else:
        left = mid + 1

最后在写check(x)函数：

def check(x):
    count = 0  # 记录我们切题的次数
    sum1 = 0  # 记录我们输入代码的行数
    for i in range(l):
        sum1 += OJ[i]
        if sum1 >= x:  # >= n，符合条件，直接AC切下一题
            sum1 = 0
            count += 1  # 切题次数加一
        if sum1 < 0:  # 删除后代码行数<0，代表代码为0行
            sum1 = 0
    return count

完整代码：

l, k = (int(x) for x in input().split(' '))
OJ = []
min_ans = 0
max_ans = 0
for i in range(l):
    OJ.append(int(input()))


def check(x):
    count = 0  # 记录我们切题的次数
    sum1 = 0  # 记录我们输入代码的行数
    for i in range(l):
        sum1 += OJ[i]
        if sum1 >= x:  # >= n，符合条件，直接AC切下一题
            sum1 = 0
            count += 1  # 切题次数加一
        if sum1 < 0:  # 删除后代码行数<0，代表代码为0行
            sum1 = 0
    return count


left = 1
right = sum(abs(i) for i in OJ)  # 定义右边界，可以定义为int(1e9)，但这会导致部分数据TLE，所以我们先求出数组的绝对值和
while left <= right:  # 求最大值
    mid = (left + right) // 2
    if check(mid) >= k:  # 切题次数太多，说明此答案较小
        left = mid + 1  # 移动右边界
        if check(mid) == k:
            max_ans = mid
    else:  # 切题次数太少，说明此答案较大
        right = mid - 1  # 移动左边界

left = 1
right = max_ans  # 右边界设为我们刚求的最大值，减少循环次数
while left <= right:  # 求最小值
    mid = (left + right) // 2
    if check(mid) <= k:
        right = mid - 1
        if check(mid) == k:
            min_ans = mid
    else:
        left = mid + 1


if max_ans == 0 or min_ans == 0:
    print(-1)
else:
    print(min_ans, max_ans)

唯一ACPython代码地址：https://www.luogu.com.cn/record/96187869