2019头条笔试——国庆旅游景点选择（python）

题目描述

小明国庆节来北京玩，北京有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

解题思路

a[i]+a[j]+i-j可以分组为(a[i]+i)+(a[j]+j)

1. 暴力解法——枚举（i，j）的组合

n = int(input())
lst = list(map(int,input().split()))
res = 0
for j in range(n):
    for i in range(j):
        res = max(res,(lst[i]+i)+(lst[j]-j))
print(res)

2. 对于每一个确定的j，只需要找到j以前最大的(a[i]+i)值即可

n = int(input())
lst = list(map(int,input().split()))
res = 0
max_score = lst[0]
for j in range(1,n):
    res = max(res,max_score + lst[j]-j)
    max_score = max(max_score,lst[j]+j)
print(res)