BOJ: 2721 삼각수의 합

문제

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

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

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

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

입력

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

출력

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

예제 입력 1

4
3
4
5
10

예제 출력 1

45
105
210
2145

코드

#include <bits/stdc++.h>

using namespace std;

int main(void) {
    int t;

    cin>>t;

    for(int i=0; i<t; i++) {
        int n;
        int sum = 0;

        cin>>n;

        for(int j=1; j<=n; j++)
            sum += (j*(j+1)*(j+2)/2);           

        cout<<sum<<'\n';     
    }

    return 0;
}

Reference

BOJ