华为OD-2024年C卷D卷-贪吃的猴子[200分]Python3实现100%通过率

题目描述

一只贪吃的猴子，来到一个果园，发现许多串香蕉排成一行，每串香蕉上有若干根香蕉。每串香蕉的根数由数组numbers给出。

猴子获取香蕉，每次都只能从行的开头或者末尾获取，并且只能获取N次，求猴子最多能获取多少根香蕉。

输入描述

第一行为数组numbers的长度

第二行为数组numbers的值每个数字通过空格分开

第三行输入为N，表示获取的次数

输出描述

按照题目要求能获取的最大数值

备注

• 1 ≤ numbers.length ≤ 100000
• 1 ≤ numbers ≤ 100
• 1 ≤ N ≤ numbers.length

用例1

输入

7
1 2 2 7 3 6 1
3

输出

10

说明

第一次获取香蕉，无论是从行的开头或者末尾获取，得到的香蕉根数目为1, 但是，从行末尾获取能获取到最优的策略，后面可以直接得到香蕉根数目6和3。因此最终根数为1+6+3=10

用例2

输入

3
1 2 3
3

输出

6

说明

全部获取所有的香蕉，因此最终根数为1+2+3=6

用例3

输入

4
4 2 2 3
2

输出

7

说明

第一次获取香蕉为行的开头，第二次获取为行的末尾，因此最终根数为4+3=7

n = int(input())
arr = list(map(int, input().split()))
k = int(input())
n = len(arr)
if k >= n:
    print(sum(arr))
    exit(0)
s = [0] * (n + 1)
for i in range(n):
    s[i + 1] = s[i] + arr[i]
mx = s[k]
now = 0
for i in range(1, k + 1):
    now += arr[n - i]
    mx = max(mx, now + s[k - i])
print(mx)
# 7
# 1 2 2 7 3 6 1
# 3

# 10

# 3
# 1 2 3
# 3

# 6

# 4
# 4 2 2 3
# 2

# 7