백준 / 1339 단어 수학 python3

www.acmicpc.net/problem/1339

1339번: 단어 수학

주어진 문자열에 숫자 0~9를 대입하여 합이 최대로 나오게 만들면 된다. 

긴 문자열 순으로 앞에서부터 9부터 넣어가면 된다.

 

하지만 하나 주의할 점은 BAA, AAA 와 같은 경우 A에 9를 넣어야 하지만, 순서대로 하면 B에 9를 넣을 수도 있다. 

그래서 우선 각 알파벳 별로 dict를 이용해 가능한 10의 거듭제곱으로 값을 만들어주었다. AAA의 경우 dict['A']에 111을 더하게 된다. 

그리고 여기서 가장 큰 값부터 9부터 곱해서 더해주면 된다.

 

코드는 다음과 같다.

n = int(input())
arr = []
for i in range(n):
    alphabet = input()
    arr.append(alphabet)

check = dict()
    
for alpha in arr:
    cnt = 0
    for i in alpha:
        if i in check:
            check[i] += 10 ** (len(alpha) - cnt - 1)
        else:
            check[i] = 10 ** (len(alpha) - cnt - 1)
        cnt += 1

val = list(check.values())
val.sort(reverse=True)

result = 0
tmp = 9
for i in val:
    result += tmp * i
    tmp -= 1

print(result)