华为机考真题 -- 贪吃的猴子

题目描述：

一只贪吃的猴子，来到一个果园，发现许多串香蕉排成一行，每串香蕉上有若干根香蕉。每串香蕉的根数由数组 numbers 给出。猴子获取香蕉，每次都只能从行的开头或者末尾获取，并且只能获取 N 次，求猴子最多能获取多少根香蕉。

输入描述：

第一行为数组 numbers 的长度
第二行为数组 numbers 的值每个数字通过空格分开
第三行输入为 N，表示获取的次数

输出描述：

按照题目要求能获取的最大数值
补充说明：
1<= numbers.length <= 100000
1<= numbers[i] <= 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

C++源码：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
	
	int numbers;
	cin >> numbers;  // 输入香蕉串的总数
	vector<int> ba(numbers);
	for (int i = 0; i < numbers; ++i) {
		cin >> ba[i]; 
	}
	ba.insert(ba.begin(), 0);  

	vector<int> total(numbers + 1, 0);  
	for (int i = 1; i <= numbers; ++i) {
		total[i] = total[i - 1] + ba[i];  // 计算前缀和，total[i]为前i个香蕉的总数量
	}

	int N;
	cin >> N;  
	int maxBa = 0; 
	for (int i = 0; i < N; ++i) {  
		int left = N - i;  // 计算从后端取香蕉的次数
		int temp = total[i] + total[numbers] - total[numbers - left];
		maxBa = max(maxBa, temp);
	}
	cout << maxBa << endl;  // 输出最大香蕉数

	system("pause");
	return 0;
}