[BOJ] 1339번 단어 수학 / 사용언어 : 파이썬(python)

※ 문제링크

1339번: 단어 수학

해당 문제는 그리디 알고리즘과 브루스포트 알고리즘과 관련된 문제로 둘 중 1개만 활용해도 풀 수 있는 문제였다. 브루스 포트 알고리즘은 모든 경우의 수를 탐색해야하기에 시간제한이 빠듯할 것 같다 생각하여 그리디 알고리즘 쪽에 초점을 맞춰 논리성을 찾는데 보다 주안점을 두고 해결하였다. 해결을 위해 사용한 논리적 조건은 아래와 같다.

1. 입력받은 알파벳별로 자릿수를 분류하고, 알파벳들이 등장할 때마다 자릿수를 기록하여 더해준다.

2. 입력값들에 대해 알파벳별 자릿수 기록을 작성한 후, 값들을 비교하여 해당 값이 큰 알파벳부터 큰 숫자를 부여해준다.

3. 입력받은 알파벳 단어들(예시: ABC)을 각각의 알파벳들에 부여된 숫자로 바꿔준다.

위의 조건들을 코드로 구현하여 해결하였으며, 자세한 풀이방법과 코드는 아래와 같다.

1. 입력받은 알파벳 문자를 알파벳 고유값, 알파벳 단어로 분리하여 리스트에 저장해준다.

2. 알파벳 문자를 반복문을 활용하여 알파벳들의 고유값을 확인하고, 해당 고유값을 모두 더해서 저장해준다.

3. 입력받은 문자들의 입력이 끝난 후에 기록된 값들 중 가장 큰 값을 가지는 알파벳부터 큰 숫자를 부여해준다.

4. 입력받은 알파벳 단어들에 알파벳별로 부여된 숫자들로 변환하고 모든 값들을 더해서 결과를 출력해준다.

 

import sys

alpha_cases = []
# input = sys.stdin.readline
N = int(input())
alphas = []
alphas_order = []
for _ in range(N):
    inputValue = list(input())
    # inputValue = list(input().rstrip())
    alpha_cases.append(inputValue)
    alphaCheck = set(inputValue)
    for alpha in alphaCheck:
        if alpha not in alphas_order:
            alphas.append([alpha, 0, 0])
            alphas_order.append(alpha)
    for i in range(1, len(inputValue)+1):
        alphas[alphas_order.index(inputValue[-i])][1] += 10**(i-1)

alphas.sort(key = lambda x: x[1], reverse=True)
alphas_order = [] # alphas와 순서를 맞춰주기 위해 새로 작성
for alpha in alphas:
    alphas_order.append(alpha[0])
    
for i in range(len(alphas)):
    alphas[i][2] = 9 - i

ans = 0
for alpha_case in alpha_cases:
    r1 = ''
    for alpha in alpha_case: 
        r1 += str(alphas[alphas_order.index(alpha)][2])
    ans += int(r1)
print(ans)

 

P.S 더 나은 개발자가 되기위해 공부중입니다. 잘못된 부분을 댓글로 남겨주시면 학습하는데 큰 도움이 될 거 같습니다.