[백준] 1339. 단어 수학 - 파이썬

[Gold IV]

 

https://www.acmicpc.net/problem/1339

1339번: 단어 수학

 

풀이

딱 봐도 파이썬으로 풀면 편하겠다 싶어서 파이썬으로 조진(?) 문제.

단어를 받아오고, 자릿수에 따른 가중치(?)를 계산 후 내림차순으로 정렬하여

Greedy하게 숫자를 배정해 주었다.

AC.

N = int(input())
 
A = dict()
words = []
 
for _ in range(N):
    word = list(input())
    words.append(word)
    word = list(reversed(word))
    for i, w in enumerate(word):
        if A.get(w):
            A[w] += pow(10, i)
        else:
            A[w] = pow(10, i)
 
A = sorted(A.items(), key=lambda x: x[1], reverse=True)
 
res = 0
for i, a in enumerate(A):
    res += a[1]*(9-i)
 
print(res)