[BOJ, Python] 1157번 : 단어 공부

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

1157번: 단어 공부

문제

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

입력

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

출력

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

예제 입력 1

Mississipi

예제 출력 1

?

예제 입력 2

zZa

예제 출력 2

Z

예제 입력 3

z

예제 출력 3

Z

예제 입력 4

baaa

예제 출력 4

A

 

풀이

생각보다 막혔던 문제였다. 여러 방법으로 접근을 해봤는데, 딕셔너리도 썼다가 집합도 썼다가 맞추지는 못했다.

 

내가 생각했을 때 이 문제의 핵심은 집합을 활용한 중복 제거와, 최빈 단어의 index를 찾는 것이다.

 

코드는 다음과 같다.

string = input().upper()
char_list = list(set(string))
cnt_list = [ string.count(char) for char in char_list ]

if cnt_list.count(max(cnt_list)) > 1:
    print("?")
else:
    max_index = cnt_list.index(max(cnt_list))
    print(char_list[max_index])

 

cnt_list에 string에 포함된 각 알파벳이 등장하는 횟수를 저장한다. 만약 최댓값이 2회 이상 등장한다면, 최빈 단어가 여러 개 존재하는 것이므로 ?를 출력한다.

 

그게 아니라면(else:) char_list와 cnt_list의 index가 지칭하는 char이 같다는 것을 활용하여, 최대 빈도 단어의 index를 max_index에 저장하고 char_list에서 그 index가 지칭하는 알파벳을 출력한다.