[SWEA #8016] 홀수 피라미드(C++)

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWvzGUKKPVwDFASy 

SW Expert Academy

#include <iostream>
#include <vector>

using namespace std;
int main()
{
	int N;
	cin >> N;
	vector<int>input(N);
	for (int i = 0; i < N; i++)
		cin >> input[i];
	for (int i = 0; i < N; i++)
	{
		long long n = input[i];
		cout << "#" << i + 1 << " " << 2 * n * n - 4 * n + 3 << " " << 2 * n * n - 1 << "\n";
	}
	return 0;

}

일반항을 구해서 제일 왼쪽은 2*n*n-4*n+3, 제일 오른쪽은 2*n*n-1으로 출력하였다.

 

문제의 왼쪽 수만 보면 1,3,9,19…다. 계차수열이다.

1 3 9 19 33

+2, +6, +10, +14. 4만큼 늘어나고 있다.

 

수열 An의 계차수열 Bn이라 한다면,

Bn=2+4(n-1)이다.

이것을 바탕으로 An의 일반항을 구할 수 있다.

 

 

https://www.mathfactory.net/10916