백준 1157번 파이썬 풀이 : 단어 공부

문제 링크 : https://www.acmicpc.net/problem/1157

1157번: 단어 공부

문제

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

입력

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

출력

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

해당 문제를 푸는 이유는 금일 진행했던 2021 Toss Next 코딩테스트에 출제된 문제 유형과 비슷한 유형이라 풀이를 올리며 다시한번 정리해 본다.

 

문제 풀이

# 소문자와 대문자의 관계없이 갯수를 세어야 하고 출력을 대문자로 하기 때문에 먼저

# 모든 문자열을 대문자로 바꿔준다.

# 후에 set과 list 를 사용하여 중복 제거하고 해당 단어의 갯수를 세어 count 배열에 저장한다.

word = input()
word = word.upper()

입력한 단어들을 대문자로 바꾸어 준다.

word_list = list(set(word))
count = []
 
for i in word_list:
    count.append(word.count(i))

word_list 에 중복된 단어들을 모두 제거하고 count 해야하는 단어들만 남긴다.

word.count( 타겟 )을 입력할 경우 word 문자열에서 타겟에 해당하는 문자열의 갯수를 리턴한다.

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

count 배열에서 최대 값을 가진 index를 index 변수에 저장한다.

만일 최대 값이 2개 이상이면 지문에 따라 "?"를 리턴하고, 아닐경우 미리 선언한 index를 출력한다.

 

위 문제와 비슷하게 Toss next 프로그래밍에 출제가 되었다.

쉽게 풀어 맞추긴 했지만 list 와 set을 적절하게 잘 사용을 할 줄 알아야 할 것 같다.