[구현/수학] 백준 2721 삼각수의 합 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

2721번: 삼각수의 합

 

 

 

2. 문제 풀이

W(n) = Sum[k=1..n; k*T(k+1)]

 T(n)은 1부터 n까지의 합입니다. 

 

 

3. 코드

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    n = int(input())
    res = sum(k*sum(range(k+2)) for k in range(1, n+1))
    print(res)

 1부터 n까지의 합은 sum(range(k+2))로 구현할 수 있습니다. 물론 등차수열의 합 공식을 사용해도 됩니다.