🐣 36진수

1. 문제 링크

1036번: 36진수

2. 코드

Python3 34148KB 72ms

# time : 50'

import sys
from collections import defaultdict

input = sys.stdin.readline
n = int(input())
str_list, num_list = [], []
num_cnt = defaultdict(int) # 숫자 중요도 횟수 체킹 딕셔너리

'''
문자열을 36진수 숫자 리스트 형태로 리턴 & 중요도 체킹
- 중요도 : 자릿수 + {'Z' - 타겟값}
'''
def str_to_notation(num_cnt, target):
    rst = []
    for i in range(len(target)):
        if (target[i].isdigit()):
            t = int(target[i])
        else:
            t = (ord(target[i]) - ord('A')) + 10
        num_cnt[t] += pow(36, len(target) - 1 - i) * (35 - t)
        rst.append(t)
    return rst


''' 숫자 리스트를 최종 문자열로 리턴 '''
def num_list_to_str(num_list):
    rst = 0 # 36진수 숫자 형태들의 전체 합
    fin_str = ''
    for i in range(n):
        target = 0
        for j in range(len(num_list[i]) - 1, -1, -1):
            target = target + num_list[i][j] * \
                pow(36, len(num_list[i]) - 1 - j)
        rst += target

    while rst: # 36진수에 따른 문자 변환
        t = rst % 36
        if (t < 10):
            fin_str = str(t) + fin_str
        else:
            fin_str = chr(t - 10 + ord('A')) + fin_str
        rst //= 36
    return fin_str if fin_str else '0'  # 0일 때 공백 리턴 exception [TC 87%]


for _ in range(n):
    str_num = input().rstrip()
    while (str_num[0] == '0' and len(str_num) > 1):  # trim 0 처리 exception [TC 87%]
        str_num = str_num[1:]
    notation = str_to_notation(num_cnt, str_num)
    num_list.append(notation)

k = int(input())

# 중요도에 따른 정렬
list_cnt_notation = list(num_cnt.items())
list_cnt_notation.sort(key=lambda x: x[1], reverse=True)

for i in range(min(k, len(list_cnt_notation))):
    t = list_cnt_notation[i][0]
    for j in range(n):
        for l in range(len(num_list[j])):
            if num_list[j][l] == t:
                num_list[j][l] = 35  # Z로 바꿔주기


print(num_list_to_str(num_list))