백준] 2721 - 삼각수의 합(ACM-ICPC Regionals)

시간 제한 : 1초

메모리 제한 : 128MB

입력

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

출력

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

소스코드

#include <iostream>
#include <vector>
using namespace std;
int main(void)
{
	vector<int> Tri(302, 0);
	int T;
	cin >> T;

	for (int i = 1; i <= 301; i++)
		Tri[i] = Tri[i - 1] + i;
	while (T--)
	{
		int n;
		cin >> n;

		int sum = 0;
		for (int k = 1; k <= n; k++)
			sum += k * Tri[k + 1];
		
		cout << sum << '\n';
	}
}

Tip

문제에서 어려운 부분은 없다. 배열에 값을 잘 넣고, 주어진 식대로 계산만 잘하면 되는 문제다.