알고리즘

[백준] DP - 2225번, 합분해(다른 풀이 봄)

hatch 2023. 11. 11. 23:41
728x90
반응형

* 공부 목표

DP - 1463, 11726, 11727, 9095, 10844, 11057, 2193, 9465, 2156, 11053, 11055, 11722, 11054, 1912, 2579, 1699, 2133, 9461, 2225, 2011, 11052

문) 0부터 N까지의 정수 K개를 더해서 그 합이 N이 되는 경우의 수를 구하는 프로그램을 작성하시오.

덧셈의 순서가 바뀐 경우는 다른 경우로 센다(1+2와 2+1은 서로 다른 경우). 또한 한 개의 수를 여러 번 쓸 수도 있다.

입) 첫째 줄에 두 정수 N(1 ≤ N ≤ 200), K(1 ≤ K ≤ 200)가 주어진다.

출) 첫째 줄에 답을 1,000,000,000으로 나눈 나머지를 출력한다.


풀이

1. 테이블 정의

  • dp[k][n] = k개의 수를 더해 n을 만들 수 있는 경우의 수

 

2. 점화식

0~N까지의 숫자가 K개 주어진다는 상황을 가정해보자.

N = 마지막 값 + 이전 값들의 누적

점화식으로 적으면, DP[k][n] = DP[k-1][0] + .. + DP[k-1][n] 이다.

 

k (하단) / n 0 1 2 3
1 dp[1][0] = 1 dp[1][1] = 1 dp[1][2] = 1 dp[1][3] = 1
2 dp[2][0] = 1 dp[2][1] = 2 dp[2][2] = 3 dp[2][3] = 4
3 dp[3][0] = 1 dp[3][1] = 3 dp[3][2] = 6 dp[3][3] = 10

 

위의 내용을 코드로 바꾸면 3중 for문으로 작성하게 된다.

더 나은 방식이 없을까?

있다.

 

위의 표 내용을 자세히 보자.

기존의 생각이 전체 값을 더한다는 것이었다면,

이제는 필요한 값만 더해온다는 관점으로 바라보자.

 

dp[2][2]를 2가지로 표현해 볼 것이다.

1) dp[1][0] + dp[1][1] + dp[1][2]

k (하단) / n 0 1 2 3
1 1 1 1 1
2 1 2 3 4
3 1 3 6 10

기존 관점과 동일하다. 표시한 파란색 값을 다 더해야 빨간색 값이 나온다.

 

2) dp[2][1] + dp[1][2]

k (하단) / n 0 1 2 3
1 1 1 1 1
2 1 2 3 4
3 1 3 6 10

 

dp[2][1] = dp[1][0] + dp[1][1]

그러므로, dp[2][2] = dp[2][1] + dp[1][2] 로 대체하여 표현할 수 있는 것이다.

 

따라서 최종 점화식은 다음과 같다.

dp[k][n] = dp[k-1][n] + dp[k][n-1]

 

 

3. 초기값

  • n = 0, dp[k][n] = 1 →dp[k][0] = 1
  • k = 1, dp[k][n] = 1 →dp[1][n] = 1

 

코드

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

public class Main {
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());

        int[][] dp = new int[k+1][n+1];

        for(int i=0; i<k+1; i++) {
            dp[i][1] = i;
        }
        for(int i=0; i<n+1; i++) {
            dp[1][i] = 1;
        }
        for(int i=2; i<k+1; i++) {
            for(int j=2; j<n+1; j++) {
                dp[i][j] = (dp[i-1][j] + dp[i][j-1])% 1000000000;
            }
        }
        br.close();
        bw.write(String.valueOf(dp[k][n]));
        bw.flush();
        bw.close();
    }
}

파도반 수열 문제 풀고, 아 이제 좀 성장했나 싶었는데 바로 막혔다.

난이도 보니까 골드Ⅴ 문제였다. 흑.

열심히 해야겠다는 다짐을 오늘도 한다.

 

작다고만 생각말고 작은 생각을 매일 키워라.

 

 


[참고] https://hongjw1938.tistory.com/63  (풀이 정말 좋다!)

https://computer-choco.tistory.com/545  (위의 내용을 이해시켜주는 좋은 풀이!)