[백준] 1157 - 단어 공부

문제

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

입력

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

출력

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

---

from collections import Counter


def solution():
    word = input().lower()
    chr_counter = Counter(word)
    chr_nums = chr_counter.most_common()

    if len(chr_nums) == 1:
        print(chr_nums[0][0].upper())
    elif chr_nums[0][1] > chr_nums[1][1]:
        print(chr_nums[0][0].upper())
    else:
        print("?")


if __name__ == "__main__":
    solution()

해설

• 단어를 문자열로 입력받고 모두 소문자로 변환한다.
• Counter 모듈을 사용해 알파벳 별 개수를 구한다.
• 만약 알파벳 개수가 하나 뿐이라면 바로 해당 알파벳을 대문자로 변환하여 출력한다.
• 둘 이상이라면 첫번째 알파벳과 두번째 알파벳의 개수를 비교해 첫번째 알파벳의 개수가 더 크면 해당 알파벳을 대문자로 변환하여 출력하고, 아니라면 "?"를 출력한다.
  • most_common 함수는 원소 별로 카운팅하여 (원소, 원소의 개수)가 내림차순으로 정렬된 리스트를 반환한다.