[백준] 단계별 문제풀이 #1157- 단어 공부

문제

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

입력

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

출력

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

풀이

import sys

str_input = sys.stdin.readline().rstrip().upper()

# unique 한 문자들을 list에 담는다
unique_list = list(set(str_input))

count_list = []

# unique 한 문자들이 input으로 받은 문자열에 몇개 있는지 count해서 List 에 담는다
for char in unique_list:
    count_list.append(str_input.count(char))

# 최대값이 count 한 list 에 몇개 포함되어 있는지 확인하고, 최대값이 유일하다면 index 값으로 그 문자를 출력한다
if count_list.count(max(count_list)) > 1:
    print("?")
else:
    print(unique_list[count_list.index(max(count_list))])

검토

첫 시도는 문자와 문자열에 포함된 문자의 개수를 Dictionary 변수에 담으니, 메모리도 많이 잡아 먹고, 시간도 오래걸렸다.
두 번째 시도에 list 변수를 사용하니 성능이 개선되었다.

출처

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