코딩 트레이닝

백준 9095번 - 1, 2, 3 더하기(C++)

Bum_2 2023. 7. 18. 11:05

문제

정수 4를 1, 2, 3의 합으로 나타내는 방법은 총 7가지가 있다. 합을 나타낼 때는 수를 1개 이상 사용해야 한다.

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

입력

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

출력

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

예제 입력 1

3
4
7
10

예제 출력 1

7
44
274

문제 풀이


#include <iostream>
using namespace std;

// 동적 계획법을 사용
int count(int n) {
    int dp[12] = {0}; // 1부터 11까지의 경우를 저장할 배열

    // 초기값 설정
    dp[1] = 1;
    dp[2] = 2; //1 + 1, 2
    dp[3] = 4; // 1+1+1, 1+2, 2+1, 3.
               // dp[3] + dp[2] + dp[1];

    for (int i = 4; i <= n; i++) {
        dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3];
    }

    return dp[n];
}

int main() {
    int T;
    cin >> T;

    while (T--) {
        int n;
        cin >> n;
        cout << count(n) << endl;
    }

    return 0;
}

dp[1] = 1
dp[2] = 2 // 1+1, 2
dp[3] = 4 // 1+1+1, 1+2, 2+1, 3
dp[i] = dp[i - 1] + dp[i - 2] + dp[i - 3]
=> dp[i-1] 에서 1 더하는 경우 + dp[i-2]에서 2 더하는 경우 + dp[i-3]에서 3 더하는 경우 (단, i > 3)