1번부터 N번까지의 학생들은 각각 블록들을 가지고 있다. 학생마다 최대 M개의 블록을 가지고 있을 수 있으며, 한 명의 학생이 가지고 있는 모든 블록들의 높이는 서로 다르다. 이 때 1번부터 N번까지의 학생들이 가진 블록을 차례대로 사용하여 바닥에서부터 쌓아올려 하나의 탑을 만들고자 한다.
단, 어떤 학생의 블록은 사용하지 않아도 되며 한 학생당 최대 1개의 블록만을 사용할 수 있다.
1번부터 N번까지의 학생들이 가지고 있는 블록들에 대한 정보가 주어졌을 때, 높이가 정확히 H인 탑을 만들 수 있는 경우의 수를 계산하는 프로그램을 작성하시오.
예를 들어 N=3, M=3, H=5일 때, 각 학생마다 가지고 있는 블록들의 높이가 다음과 같다고 가정하자.
이 때, 탑의 높이가 정확히 5가 되도록 블록을 쌓는 경우로는 다음의 6가지가 존재한다. (블록을 사용하지 않는 경우는 X로 표시하였다.)
첫째 줄에 자연수 N, M, H가 공백을 기준으로 구분되어 주어진다. (1 ≤ N ≤ 50, 1 ≤ M ≤ 10, 1 ≤ H ≤ 1,000) 둘째 줄부터 N개의 줄에 걸쳐서 각 학생이 가진 블록들의 높이가 공백을 기준으로 구분되어 주어진다.
단, 모든 블록의 높이는 1,000 이하의 자연수이며 한 명의 학생이 가지고 있는 모든 블록들의 높이는 서로 다르게 주어진다.
첫째 줄에 높이가 H인 탑을 만드는 경우의 수를 10,007로 나눈 나머지를 출력한다.
3 3 5
2 3 5
3 5
1 2 3
6
이 문제는 다이내믹 프로그래밍 알고리즘을 이용해서 풀 수 있었다.
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
public class Main {
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] input = br.readLine().split(" ");
int N = Integer.parseInt(input[0]);
int H = Integer.parseInt(input[2]);
int[][] dp = new int[N+1][H+1];
ArrayList<Integer>[] list = new ArrayList[N+1];
for(int i=1; i<=N; i++)
list[i] = new ArrayList<>();
for(int i=1; i<=N; i++) {
input = br.readLine().split(" ");
for(int j=0; j<input.length; j++)
list[i].add(Integer.parseInt(input[j]));
}
for(int i=0; i<=N; i++) //0일 가능성은 모두 1이기 때문에
dp[i][0] = 1;
for(int i=1; i<=N; i++) {
for(int j=1; j<=H; j++) {
for(int x : list[i]) {
if(j>=x) {
dp[i][j] += dp[i-1][j-x] %= 10007; //해당 블록을 쌓을수 있는 경우 더해줌
}
}
dp[i][j] += dp[i-1][j] %= 10007; //블록을 안쌓을 경우
}
}
System.out.println(dp[N][H]%10007);
}
}