[백준/9095] 1, 2, 3 더하기 - Java 자바

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

 

개념: dp

 

문제

정수 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 = 4일때를 예시로 보면, `dp[N] = dp[N-1] + dp[N-2] + dp[N-3]`으로 나타낼 수 있음을 알 수 있다.

3+1, 2+1+1, 1+2+1, 1+1+1+1
2+2, 1+1+2
1+3

 

따라서 초기값은 dp[1], dp[2], dp[3]까지 설정해준다.

 

 

전체코드

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine());
        int[] problem = new int[T];
        for (int t = 0; t < T; t++) {
            int N = Integer.parseInt(br.readLine());
            problem[t] = N;
        }
        for (int i = 0; i < T; i++) {
            int n = problem[i];
            int[] dp = new int[n+5];
            dp[1] = 1; dp[2] = 2; dp[3] = 4;
            for (int j = 4; j <= n; j++) {
                dp[j] = dp[j-1] + dp[j-2] + dp[j-3];
            }
            System.out.println(dp[n]);
        }
    }
}