18427 함께 블록 쌓기

정민용·2023년 5월 3일

백준

목록 보기
174/286

문제

1번부터 N번까지의 학생들은 각각 블록들을 가지고 있다. 학생마다 최대 M개의 블록을 가지고 있을 수 있으며, 한 명의 학생이 가지고 있는 모든 블록들의 높이는 서로 다르다. 이 때 1번부터 N번까지의 학생들이 가진 블록을 차례대로 사용하여 바닥에서부터 쌓아올려 하나의 탑을 만들고자 한다.

단, 어떤 학생의 블록은 사용하지 않아도 되며 한 학생당 최대 1개의 블록만을 사용할 수 있다.

1번부터 N번까지의 학생들이 가지고 있는 블록들에 대한 정보가 주어졌을 때, 높이가 정확히 H인 탑을 만들 수 있는 경우의 수를 계산하는 프로그램을 작성하시오.

예를 들어 N=3, M=3, H=5일 때, 각 학생마다 가지고 있는 블록들의 높이가 다음과 같다고 가정하자.

  • 1번 학생: 2, 3, 5
  • 2번 학생: 3, 5
  • 3번 학생: 1, 2, 3

이 때, 탑의 높이가 정확히 5가 되도록 블록을 쌓는 경우로는 다음의 6가지가 존재한다. (블록을 사용하지 않는 경우는 X로 표시하였다.)

# 18427
import sys
input = lambda: sys.stdin.readline().strip()

n, m, h = map(int, input().split())
people = [sorted(list(map(int, input().split()))) for _ in range(n)]
knapsack = [[0] * (h+1) for _ in range(n+1)]

for i in range(1, n+1):
    for j in range(h+1):
        knapsack[i][j] += knapsack[i-1][j]
    
    for b in range(len(people[i-1])):
        block = people[i-1][b]
        if block > h:
            break
        knapsack[i][block] += 1
        
        for j in range(h+1):
            if block > j:
                continue
            else:
                value = knapsack[i-1][j-block]
                knapsack[i][j] += value
                
                knapsack[i][j] = knapsack[i][j] % 10007
                    
for k in knapsack[::-1]:
    if k[h] != 0:
        print(k[h])
        exit(0)
print("0")

백준 18427 함께 블록 쌓기

0개의 댓글