[백준] 1157 단어 공부

문제

알파벳 대소문자로 된 단어가 주어지면, 이 단어에서 가장 많이 사용된 알파벳이 무엇인지 알아내는 프로그램을 작성하시오. 단, 대문자와 소문자를 구분하지 않는다.

입력

첫째 줄에 알파벳 대소문자로 이루어진 단어가 주어진다. 주어지는 단어의 길이는 1,000,000을 넘지 않는다.

출력

첫째 줄에 이 단어에서 가장 많이 사용된 알파벳을 대문자로 출력한다. 단, 가장 많이 사용된 알파벳이 여러 개 존재하는 경우에는 ?를 출력한다.

 

처음 쓴 코드

word = input().upper()

alphabet_count = {}
for w in word:
    alphabet_count[w] = alphabet_count.get(w, 0) + 1 # 알파벳 개수 카운팅

M = [k for k, v in alphabet_count.items() if v == max(alphabet_count.values())]
if len(M) >= 2:
    print('?')
else:
    print(M[0])

 

코딩테스트용 문법으로 수정

max_value = max(alphabet_count.values()) # 이전에 한 건 매 반복마다 max()를 계산함. 따라서 이런 상수는 효율을 위해 따로 미리 구해주는 것이 더 좋다. 
M = [k for k, v in alphabet_count.items() if v == max_value]

몇 년 전 맞췄던 정답을 확인해보니까 지금 이 코드의 실행 속도가 훨씬 빠른 것을 확인했다.

그래서 내가 그때 그런 코드들로 코테를 떨어졌구나 싶었다.