PS1 - Basic Python (05-12 Oct)
2. Fluortanten
题目是给一个全部为数字,只包含一个0的list,随意移动0的位置,使得list中的每个数和他的位置相乘的和最大。
首先对输入进行处理
num = input()
lst_tmp = input()
lst_tmp = lst_tmp.split(' ')
split以后需要把所有数字转化成iint的形式
这里使用map()
直接使用map会报错:
map(int, lst_tmp)
>> <map object at 0x104a4ef70>
这里需要map之后转成list:
lst = list(map(int, lst_tmp))
print(lst)
>> [1, 2, 3, 4]
完整代码:
num = input()
lst_tmp = input()
lst_tmp = lst_tmp.split(' ')
lst = list(map(int, lst_tmp))
lst.remove(0)
# 先算出来lst中去掉0以后的全部的和
sum1 = 0
for idx, item in enumerate(lst):
sum1 += (idx + 1) * item
# 再从后往前加,得到和的最大值
max_tmp = -1e9
sum2 = 0
for i in lst[::-1]:
sum2 += i
if sum2 > max_tmp:
max_tmp = sum2
# 最后判断和的最大值是否比0大,如果大就与原始lst的计算结果相加
if max_tmp > 0:
sum1 += max_tmp
# print出结果
print(sum1)