[백준 알고리즘][파이썬/Python] 1157번: 단어 공부

문제

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

 

입력

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

 

출력

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

 

입출력 예

 

제출 코드

s = input().upper()
dic = {}
for i in s:
    if i in dic:
        dic[i] += 1
    else:
        dic[i] = 1
best = [-1, '']
second = [-2, '']
for k, i in dic.items():
    if i >= best[0]:
        second = best
        best = [i, k]
if best[0] == second[0]:
    print('?')
else:
    print(best[1])

(1) line 1

문제에서 대문자와 소문자의 구분이 없다고 가정했기 때문에 소문자든 대문자든 하나로 통일해주는 것이 더 좋다. 다만, 출력의 형태를 대문자로 출력하라고 하였으니 대문자로 통일해주었다.

 

(2) line 2 ~ 7

입력받은 문자열에서 알파벳을 하나씩 순회하며 어느 알파벳이 몇 번 등장했는지 딕셔너리에 기록한다.

 

(3) line 8, 9

가장 많이 사용된 알파벳을 출력해주어야 하기 때문에 어떤 알파벳이 가장 많이 쓰였는지, 그 알파벳은 몇 번 등장했는지 기록하는 리스트를 선언한다. 단, '출력'에 대한 설명에서 가장 많이 사용된 알파벳이 2개 이상일 경우에는 ?를 써주라고 하였다. 나는 마땅한 아이디어가 떠오르지 않아서 두 번째로 많이 등장하는 알파벳도 기록하기 위해 second라는 리스트도 best와 같은 형태로 선언해준다.

 

(4) line 10 ~ 13

딕셔너리 내의 각 아이템을 순회하며 best과 second에 해당하는 알파벳과 그 알파벳이 등장한 횟수를 기록한다. 여기서 주의할 점은 best가 등장했다면 현재 best에 기록되어 있는 값을 second에 먼저 저장해준 뒤, best 위에 덮어써줘야 한다는 점이다.

 

(5) line 14 ~ 17

만약 best와 second에 기록되어 있는 '알파벳의 등장 횟수'가 동일하다면 '출력'에 대한 설명에서 말했듯이 ?를 출력해주어야 한다. 그렇지 않다면 best에 기록되어 있는 알파벳이 유일하게 가장 많이 사용된 알파벳이므로 해당 알파벳을 출력해준다.

 

 

 

Source : https://www.acmicpc.net/problem/1157