[Python/파이썬] 백준 16508번 전공책

16508번: 전공책

 

문제

곧 졸업을 앞둔 민호는 대학교 생활 동안 구매만 해놓고 한 번도 펴보지 않은 전공책에 먼지가 쌓여 있는 것을 보고는, 이 책들을 어떻게 처리해야 할지 고민 중이다. 열심히 고민한 끝에 민호는 결국 전공책을 모두 버리기로 마음먹었다. 하지만 그냥 버리기엔 심심한 민호는 전공책 제목에 있는 글자들을 오려서 단어 만들기 놀이를 하려고 한다. 단어 만들기 놀이는 아래 예시와 같다.

• 1번 책 : COMPUTERARCHITECTURE (35,000원)
• 2번 책 : ALGORITHM (47,000원)
• 3번 책 : NETWORK (43,000원)
• 4번 책 : OPERATINGSYSTEM (40,000원)

만약 민호가 만들고 싶은 단어가 ALMIGHTY라고 하면, 위 4개의 책 중, 1번 책에서 A를, 2번 책에서 L, M, I, G, H, T를, 4번 책에서 Y를 오려내어 원하는 단어를 만들 수 있다. 이때 드는 비용은 1번, 2번, 4번 책 가격의 합인 122,000원이다.

만약 ANT라는 단어를 만들고 싶다고 하면, 2번 책에서 A를, 3번 책에서 N, T를 오려내어 원하는 단어를 만들 수 있다. 이때 드는 비용은 2번과 3번 책 가격을 합하여 90,000원이다. 그런데, ANT라는 단어에서 A를 2번 책에서 오려내지 않고, 4번 책에 있는 A를 오려낼 수도 있다. 만약 4번 책에서 A를 오려냈을 때 드는 비용은 3번과 4번 책 가격을 합하여 83,000원으로 2번과 3번 책을 고른 비용보다 작다. 하지만, 4번 책에는 ANT가 모두 포함되어 있으므로, 4번 책만 선택했을 때 드는 비용이 40,000원이다. 이는 ANT라는 단어를 만들기 위해서 가장 적은 비용이다.

민호는 여러 개의 전공책들을 나열해 놓고는, 심각한 고민 끝에 전공책 제목에 있는 글자를 오려내어 자신이 원하는 단어를 만들기 위해서는 여러 가지 경우가 있다는 것을 깨달았다. 매우 심심했던 민호는 가장 적은 비용으로 자신이 원하는 단어를 만들려면 어떤 전공책들을 선택해야 하는지 궁금했다. 하지만 일일이 가능한 조합을 만들어 보는 것은 매우 시간 낭비라고 생각한 민호는 컴퓨터공학과답게 프로그램을 만들려고 한다.

민호를 도와 각 전공책의 가격과 제목이 주어졌을 때, 가장 적은 비용으로 민호가 원하는 단어를 만들기 위해서 어떤 전공책을 선택해야 하는지 알아보자.

입력

첫 번째 줄에는 민호가 만들고자 하는 단어를 의미하는 문자열 T (1 ≤ |T| ≤ 10)가 주어진다. T는 항상 대문자이며, |T|는 문자열 T의 길이를 의미한다.

두 번째 줄에는 민호가 가진 전공책의 개수를 의미하는 정수 N (1 ≤ N ≤ 16)이 주어진다.

다음 N개의 각 줄에는 전공책 가격을 의미하는 정수 Ci (10,000 ≤ Ci ≤ 100,000)와 제목을 의미하는 문자열 Wi (1 ≤ |Wi| ≤ 50)가 주어진다. Wi는 항상 대문자이다.

출력

민호가 원하는 단어 T를 만들기 위해서 선택해야 하는 전공책의 가장 적은 가격의 합을 출력한다. 만약 단어를 만들 수 없다면 -1을 출력한다.

코드

import sys
input = sys.stdin.readline

voca = input().rstrip()
required = [0] * 26 
# 필요한 알파벳 개수 카운트
for v in voca:
    required[ord(v)-65] += 1

n = int(input())
books = []
for i in range(n):
    cost, title = input().split()  # 가격, 제목
    books.append([title, int(cost)])
 
ans = int(1e9)
selected = [0] * 26 # 선택된 책들의 알파벳

def dfs(idx, total_cost):
    global ans
    if idx == len(books):
        # 만들 수 있는지 확인
        for i in range(26):
            if required[i] > selected[i]:
                return
        ans = min(ans, total_cost)
        return
    # 현재 책 선택 O
    for c in books[idx][0]: # c는 현재 책의 제목 글자 하나
        selected[ord(c)-65] += 1    
    dfs(idx+1, total_cost + books[idx][1])
    # 현재 책 선택 X
    for c in books[idx][0]: # c는 현재 책의 제목 글자 하나
        selected[ord(c)-65] -= 1
    dfs(idx+1, total_cost)
    
dfs(0, 0)
print(ans if ans != int(1e9) else -1)

책이 최대 16권 밖에 되지 않으므로 현재 인덱스(idx) 책을 선택함/안 함으로 나누어서 풀었다. 

required 배열에는 민호가 만드려는 단어에 필요한 알파벳의 개수를 저장해둔다. selected 배열에는 dfs가 모든 책을 탐색했을 때 선택된 책들에 포함된 알파벳의 개수가 저장되어 있는데 required와  selected 배열을 비교하여 selected 배열에 있는 알파벳들로 민호가 원하는 단어를 만들 수 있다면 이 때의 total_cost를 ans와 비교하여 둘 중 최소값을 ans에 저장해준다.