[백준 알고리즘] 백준 2721번 삼각수의 합 파이썬(Python)

@notepad_jj2

츄르사려고 코딩하는 코집사입니다.

---

1. [백준 알고리즘] 백준 2721번 삼각수의 합 파이썬(Python)

1) 문제번호 : 2721번

 

2) 문제 출처

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

2721번: 삼각수의 합

 

2. 문제

n번째 삼각수, T(n)은 1부터 n까지의 합이다. T(n) = 1 + ... + n. 이것은 삼각형 모양으로 표현할 수 있다. 아래 그림은 T(4)를 나타낸 것이다.

다음과 같은 식을 통해 가중치를 부여한 삼각수의 합을 구할 수 있다.

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

n이 주어졌을 때, W(n)을 구하는 프로그램을 작성하시오.

 

 

3. 제약사항

 

4. 입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 정수 n 하나로 이루어져 있다. (1<=n<=300)

 

 

 

5. 출력

각 테스트 케이스에 대해 W(n)을 한 줄에 하나씩 출력한다.

 

6. 풀이

- 주어진 식처럼 계산하여 출력하면 된다.

 

7. 소스 코드

import sys
input = sys.stdin.readline

T = int(input())

for i in range(T) :
    temp = int(input())

    sum = 0
    T = 0

    for j in range(1, temp+1) :
        for k in range(1, j+2) :
            T += k
        sum += j*T
        T = 0
    print(sum)

---