[Baekjoon] 1157. 단어 공부

1 minute read

단어 공부

문제 설명


문제

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

입력

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

출력

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

예제 입력 1

Mississipi

예제 출력 1

?

예제 입력 2

zZa

예제 출력 2

Z

예제 입력 3

z

예제 출력 3

Z

예제 입력 4

baaa

예제 출력 4

A


문제 풀이


collections 모듈의 Counter 클래스를 사용합니다.

Counter 생성자에 인자로 문자열을 전달하고 most_common() 메서드를 사용하면 간편하게 각 원소가 몇 개 있는 지 반환하는 리스트를 생성할 수 있습니다.

가장 개수가 많은 두 개의 원소를 뽑아 개수를 비교하고 문제에서 요구한대로 출력하는데, 문자열의 길이가 1이라면 예외처리해주어 문자열을 바로 출력하도록 합니다.

from collections import Counter

s = input().upper()
try: 
    first, second = Counter(s).most_common(2)
    print(first[0] if first[1] > second[1] else '?')
except: # 길이가 1인 경우
    print(s)


Counter 클래스

  • import 하기
from collection import Counter
  • most_common 사용법
Counter(문자열).most_common()
  • 예시
in:
  s = 'abcabcaaab'
	most_common = Counter(s).most_common()
	print(most_common)
  
out:
  [('a', 5), ('b', 3), ('c', 2)]
  
in:
  s = 'abcabcaaab'
	most_common = Counter(s).most_common(1)
	print(most_common)
  
out:
  [('a', 5)]

Leave a comment