解题思路:找到处理数据的合理方法,前缀和(说到底还是数学敏感)
注意事项:注意sum()嵌套问题,尽量减少循环
参考代码:
暴力法:itertools的组合
import itertools
n = int(input())
lst = list(map(int, input().split()))
smr = 0
for data in itertools.combinations(lst, 2) :
smr += data[0] * data[1]
print(smr)
能拿一点分,很直接的想法
未优化前的前缀和算法:
n = int(input())
lst = list(map(int, input().split()))
smr = 0
for i in range(n) :
impact -= lst[i]
smr += lst[i] * sum(lst[i+1:])
print(smr)
能拿更多的分,但是sum()复用过多
优化后的前缀和算法:
n = int(input())
lst = list(map(int, input().split()))
impact = sum(lst)
smr = 0
for i in range(n) :
impact -= lst[i]
smr += lst[i] * impact
print(smr)
就用了一次sum()
0.0分
1 人评分