Baekjoon 14929 | 귀찮아 (SIB)(Python)

📌문제

SILVER5 / 누적합 / 수학

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

설명이 왜 저런진 모르겠다..

입력은 n 그리고 xi가 각 줄로 받는다.

📌문제 분석 및 설계

사실 누적합 문제란걸 알고 들갔기에 누적합을 사용하는 방법을 탐구했다.

 

n=4일때,

(x1x2+x1x3+x1x4)+(x2x3+x2x4)+(x3x4)

=x1(x2+x3+x4)+x2(x3+x4)+x3(x4)

가 되기에, 끝에서부터 누적합을 하면 되겠다 싶었다.

 

📌소스 코드

n=int(input())
xi=list(map(int,input().split()))
prefixSum =0
result=0
for i in range(n-1,0,-1):
    prefixSum += xi[i]
    result+=prefixSum*xi[i-1]
print(result)