字节跳动2019笔试题

小明国庆节来北京玩，北京有N个景点，第 i 个景点的评分用a[i]表示，两个景点i, j之间的距离为j - i(j > i)。

小明一天只能游玩两个景点，我们认为总评分是两个景点的评分之和减去两个景点之间的距离，即为a[i]+a[j]+i-j。

那么小明选择哪两个景点才会总评分最大呢？

输入格式

第一行包含整数N。

第二行分别输入N个景点的评分。

输出格式

输出最大评分

数据范围

2≤N≤1052≤N≤105,
1≤a[i]≤10001≤a[i]≤1000

输入样例：

5
11 6 5 18 12

输出样例：

29

 

O(N^2) 做法超时

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int nums[N];

int main()
{
	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> nums[i];
	}
	int res = 0;
	for (int i = 0; i < n; i++) {
		for (int j = i+1; j < n; j++) {
			res = max(res, nums[i] + nums[j] + i - j);
		}
	}
	cout << res << endl;
}

 O(N)的做法是遍历j，每次维护一个0到j的最大值就行。

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int nums[N];

int main()
{
	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> nums[i];
	}
	int res = 0;
	int max_score = nums[0];
	for (int j = 1; j < n; j++) {
		res = max(res, max_score + nums[j]-j);
		max_score = max(max_score, nums[j] + j);
	}
	cout << res << endl;
}