백준 10539 수빈이와 수열 with Python

 

www.acmicpc.net/problem/10539

10539번: 수빈이와 수열

---

<My code>

n = int(input())
array = list(map(int, input().split()))

result = []
for i in range(n):
    temp = array[i] * (i+1)
    for i in range(len(result)):
        temp -= result[i]
    result.append(temp)
for i in result:
    print(i, end=' ')

---

<Reference Code>

n = int(input())
array = list(map(int, input().split()))

result = [array[0]]
for i in range(1, n):
    result.append(array[i] * (i+1) - sum(result))
for i in result:
    print(i, end=' ')

나는 result 리스트에 들어있는 값을 하나씩 빼주는 방법을 이용했는데,

reference code에서는 sum()을 활용해 빼는 방법을 이용했다.

이편이 코드가 간결하고 시간복잡도 면에서 더 유리하다.