第十五届蓝桥杯省赛PythonC组F题【挖矿】题解（AC）

题意分析

一条数轴上有 $n$ 个点有宝藏，你需要从 $0$ 开始，每走一格消耗一点体力，问能收集宝藏的最大值。

思路分析

根据贪心的策略，转弯次数两次及以上的方案一定不是最优解。

转弯一次和零次的方案才可能是最优解。

那么我们可以从 $0$ 点开始枚举向左（右）走到的所有位置的左侧（右侧）的价值，再加上使用剩余体力向反方向走带来的价值。

可以用前缀和优化从 $[1,x]$ 的区间价值和的计算。

N = int(2e6 + 10)
 
import sys
sys.setrecursionlimit(1000000)
input = lambda:sys.stdin.readline().strip()
 
n, m = map(int, input().split())
a = list(map(int, input().split()))
 
l = [0 for i in range(N)]
r = [0 for i in range(N)]
 
for x in a:
    if x > 0:
        r[x] += 1
    elif x < 0:
        l[-x] += 1
 
for i in range(1, N):
    l[i] += l[i - 1]
    r[i] += r[i - 1]
 
res = 0
for i in range(1, m + 1):
    s = l[i]
    if m - 2 * i > 0:
        s += r[m - 2 * i]
    res = max(res, s)
 
for i in range(1, m + 1):
    s = r[i]
    if m - 2 * i > 0:
        s += l[m - 2 * i]
    res = max(res, s)

for x in a:
    if x == 0:
        res += 1
 
print(res)

【在线测评】