본문 바로가기
Beakjoon&프로그래머스/파이썬

[백준/파이썬] 14929번 귀찮아 (SIB)

by 현장 2022. 5. 7.

-Code

n = int(input())
nums = list(map(int, input().split()))
accum = [nums[0]]
result = 0
for i in range(1, n):
    accum.append(nums[i] + accum[i - 1])

for i in range(n):
    num = nums[i] * (accum[n - 1] - accum[i])
    result += num
print(result)

처음한 작성한 것은 시간초과가 나서 찾아보니 누적합을 이용하여 푸는 방법이 있었습니다.