[Baekjoon/C++] 15990번 - 1, 2, 3 더하기 5

[백준] Baekjoon Online Judge

문제로 이동

 

문제

정수 4를 1, 2, 3의 합으로 나타내는 방법은 총 3가지가 있다. 합을 나타낼 때는 수를 1개 이상 사용해야 한다. 단, 같은 수를 두 번 이상 연속해서 사용하면 안 된다.

• 1+2+1
• 1+3
• 3+1

정수 n이 주어졌을 때, n을 1, 2, 3의 합으로 나타내는 방법의 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 정수 n이 주어진다. n은 양수이며 100,000보다 작거나 같다.

 

출력

각 테스트 케이스마다, n을 1, 2, 3의 합으로 나타내는 방법의 수를 1,000,000,009로 나눈 나머지를 출력한다.

 

---

풀이

#include <iostream>
#define MOD 1000000009
using namespace std;

long long arr[100001][3];

void df();

void df() {
    arr[1][0] = arr[2][1] = 1;
    arr[3][0] = arr[3][1] = arr[3][2] = 1;

    for (int i = 4; i < 100001; i++) {
        arr[i][0] = (arr[i - 1][1] + arr[i - 1][2]) % MOD;
        arr[i][1] = (arr[i - 2][0] + arr[i - 2][2]) % MOD;
        arr[i][2] = (arr[i - 3][1] + arr[i - 3][0]) % MOD;
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    df();

    int T;
    cin >> T;

    while (T--) {
        int n;
        cin >> n;

        cout << (arr[n][0] + arr[n][1] + arr[n][2]) % MOD << '\n';
    }

    return 0;
}