[백준] 1157번 단어 공부 | 파이썬 Python

문제

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

 

입력

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

 

출력

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

 

예제

 

나의 풀이

word = input()
count = {}

for i in word:
    str = i.upper()
    count.setdefault(str, 0)
    count[str] += 1

max_key = max(count, key=count.get)
max_value = count[max_key]
arr = list(count.values())

if arr.count(max_value) > 1:
    print("?")
else:
    print(max_key)

• `{'M': 1, 'I': 4, 'S': 4, 'P': 1}` 형태의 키는 대문자, 값은 해당 대문자의 갯수인 딕셔너리를 생성한다.
  • `count.setdefault(str, 0)`로 딕셔너리에 `str` 키가 없을 경우에만 값을 `0`으로 가지는 `str` 키를 생성한다.
• 딕셔너리에서 값이 가장 큰 키와 값을 추출한다.
• 딕셔너리를 리스트로 변환한 후에 `max_value`를 가진 요소가 몇 개인지 카운트한다.
• 만약 카운트 개수가 `1` 이상이라면 `?`를 출력하고, 아니라면 `max_key`를 출력한다.

 

다른 사람의 풀이

word = input().upper() # Mississipi
word_list = list(set(word)) # ['P', 'M', 'S', 'I']
cnt = []

for i in word_list:
    cnt.append(word.count(i)) # cnt = [4, 1, 1, 4]


max_count = max(cnt)
if cnt.count(max_count) > 1:
    print("?")
else:
    print(word_list[cnt.index(max_count)])

• `word.upper`로 단어 전체를 대문자로 만든다.
• `set()`을 사용해서 중복되는 알파벳은 제외한 리스트를 만든다.
• for문을 이용해서 `word`에 각각의 알파벳이 몇 개 들어있는지 카운트해서 `cnt`에 추가한다.
• `cnt`의 요소 중 최댓값을 구한다.
• `cnt`에 최댓값인 요소가 두 개 이상으로 중복되면 `?`을 출력하고 아니라면 최댓값에 해당하는 알파벳을 출력한다.