[Python] 백준 1157번 - 단어 공부

문제

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

입력

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

출력

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

 

 

---

정답 코드

word = input().upper()
spell = list(set(word))

cnt_list = []
for i in spell:
  cnt = word.count(i)
  cnt_list.append(cnt)

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

 

풀이

word = input().upper()
spell = list(set(word))

어차피 대문자로 출력할 것이기 때문에 처음부터 upper()를 이용해 대문자로 입력받는다.

이후 spell = list(set(word)) 를 이용하여

 

word를 set 으로 바꿔 중복 단어를 제거하고, 다시 list로 바꿔 다루기 편하게 만들어 준다.

 

cnt_list = []
for i in spell:
  cnt = word.count(i)
  cnt_list.append(cnt)

각 글자들의 갯수를 저장하기 위한 cnt_list를 선언해준다.

 

이후 중복을 없앤 글자가 담긴 spell 의 수만큼 반복을 시켜주며 

 

각 글자별 개수를 word.count(i)를 이용해 찾아서 cnt_list에 추가해준다.

 

정리하자면

word - 처음 입력받은 값 전체

spell - 입력받은 값에서 중복을 제거한 각 글자들의 리스트

cnt_list - 각 글자들의 개수를 담은 리스트

 

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

if문 부분은 cnt_list에서 가장 큰 값이 똑같은 글자가 2개 이상이라면 "?"를 추가하기 위한 조건문

 

else문은 cnt_list에서 가장큰 값을 갖고 있는 부분의 index 번호를 max_index에 저장한 후

 

spell[max_index]로 호출하여 중복된 글자가 가장 많은 글자를 출력한다.