[c++][BOJ] 파도반 수열

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

9461번: 파도반 수열

삼각형을 추가해서 그 삼각형의 변의 길이를 구하는 문제입니다.

 

123 번째는 1, 45 번째는 2의 크기를 가지고 있고, 그 이후부터는 arr[i] = arr[i-1] + arr[i-5] 가 성립합니다.

 

미리 배열을 다 만들어 놓은 후, 수가 입력될 때 마다 그 위치의 원소를 꺼내서 출력합니다.

#include <iostream>
using namespace std;

int main(){
    int n;
    cin >> n;
    
    long long arr[101];
    
    for(int i=0; i<101; i++){
        if(i == 0 || i == 1 || i == 2)
        {
            arr[i] = 1;
        }
        else if(i == 3 || i == 4)
        {
            arr[i] = 2;
        }
        else if (i >= 5)
        {
            arr[i] = arr[i-1] + arr[i-5];    
        }
    }
    
    for(int j=0; j<n; j++)
    {
        int m;
        cin >> m;
        cout << arr[m-1] << endl;
    }
    
}