头条笔试题之国庆旅行

头条笔试题之国庆旅行

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

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

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

输入格式
第一行包含整数N。

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

输出格式
输出最大评分

数据范围
2≤N≤105,
1≤a[i]≤1000
输入样例：
5
11 6 5 18 12
输出样例：
29

思路一： 暴力法，固定一天，遍历后面的天

n = eval(input())
a = list(map(eval, input().split(' ')))

max_v = 0
for i in range(n):
    for j in range(i+1, n):
        max_v = max(max_v, a[i]+a[j]+i-j)
print(max_v)

5
11 6 5 18 12
29

时间复杂度 O(n^2)
空间复杂度 O(1)
会超时的哟，需要优化的呀

思路二：改进暴力法

我们将最大值公式转换一下：

a[i]+a[j]+i-j = (a[i] + i) + (a[j] - j)

其中 j > i , 就可以固定 j， 更新 0~j-1 的最大值

记住套路，这种可以减少一个 for 循环

res = 0
max_i = a[0]
for j in range(1, n):
    res = max(res, max_i + a[j] -j)
    max_i = max(max_i, a[j] + j)
print(res)

29

时间复杂度减少为 O(n)