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

문제

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

입력

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

출력

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

---

풀이

import sys
input = sys.stdin.readline

word = str(input()).strip().upper()
word_list = list(set(word))
answer = []

for i in word_list:
    answer.append(word.count(i))

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

count를 사용하면 편하게 문자의 개수를 알 수 있음

---

초반 풀이

word = str(input()).strip().upper()
ans = {}

for i in word:
    if i not in ans:
        count = 1
        ans[i] = count
    else:
        count = ans[i] + 1
        ans[i] = count

ans_values = list(ans.values())
ans = sorted(ans.items(), key = lambda x : x[1], reverse = True)
count = 0

for i in ans:
    if(max(ans_values) != i[1]):
        continue
    else:
        count += 1

if len(ans) == 1 or count == 1:
    print(ans[0][0])
else:
    print("?")

count 함수를 몰랐을 때, 해당 처럼 굉장히 긴 코드로 작성했다,,,

 

count 함수를 알고 난 후에는 하단 코드로 수정

word = str(input()).strip().upper()
word_list = list(set(word))
answer = []
count = 0

for i in word_list:
    answer.append(word.count(i))

for i in answer:
    if max(answer) == i:
        count += 1

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

다만, 해당 코드도 count함수를 활용하여 길이를 더 줄일 수 있으므로 한 번 더 줄여 최종 코드 작성함

 

 

 

문제 출처

https://www.acmicpc.net/problem/1157