[백준 14929] 귀찮아 - python

https://www.acmicpc.net/problem/14929

14929번: 귀찮아 (SIB)

 

문제 풀이

• 처음에 어떻게 풀어야 할 지 몰라서 주위에 자문을 구해 어떤 연산을 해야하는지 알아냈다

• x1, x1+x2, x1+x2+x3, x1+x2+x3+...+xn까지의 누적합을 num에 넣고, num의 마지막 index - num의 index를 순차적으로 돌아 위 공식과 동일하게 코드가 동작할 수 있도록 했다 

n = int(input())
temp = list(map(int, input().split()))
num = [] 
num.append(temp[0])
for i in range(1, n):
    num.append(num[i-1] + temp[i])
# num = [x1, x1+x2, x1+x2+x3 ...]가 저장됨
answer = 0
for i in range(n):
    answer += temp[i] * (num[n-1] - num[i]) # x1(x2+x3) + x2(x3)형태가 됨

print(answer)