백준: 1339 단어 수학

문제

1339번: 단어 수학

 

문제 풀이

 각 알파벳이 주어질 때, 어느 자리에 위치하느냐에 따라 0에서 9의 숫자로 변환하여 합을 계산하여야 한다. 문제를 풀기 위해서 자릿수가 가장 큰 알파벳부터 9부터 0까지 숫자를 변환하면 된다. 예를 들어 예제 입력 2의 경우 {'A': 10000, 'C': 1010, 'G': 100, 'D': 100, 'E': 10, 'F': 1, 'B': 1}와 같이 A, C, G, D, E, F, B 순으로 9에서 3까지의 숫자로 변환한다. 그러면, GCF + ACDEB는 784 + 98653라는 것을 알 수 있다.

 

 문제를 풀기 위해서는 다음과 같은 로직을 따르면 쉽게 해결할 수 있다.

• defaultdict를 활용하여 각 알파벳의 자릿수를 계산한다.
• 자릿수가 가장 큰 알파벳부터 정렬하여, 9부터 숫자를 할당하여 계산한다.

 

코드

from sys import stdin
from collections import defaultdict


if __name__ == '__main__':
    answer = 0
    n = int(stdin.readline())
    alphabet = defaultdict(int)
    change = [num for num in range(10)][::-1]

    for _ in range(n):
        word = stdin.readline().strip()
        length = len(word)
        for idx, w in enumerate(word):
            alphabet[w] += 10 ** (length - idx - 1)

    alphabet = dict(sorted(alphabet.items(), key=lambda x: x[1], reverse=True))
    print(alphabet)
    for c, cnt in zip(change, alphabet.values()):
        answer += c * cnt

    print(answer)