[백준/BOJ] 1157번: 단어 공 (Python 파이썬)

문제

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

 

입출력 규칙

1. 입력

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

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

 

---

문제 풀이

<내 풀이1>

 

• set()으로 중복을 제거하여 list를 반들어 저장한 후, 알파벳의 빈도수 cnt 를 계산한다.
• cnt 리스트에서 가장 높은 빈도수를 찾음 (max(cnt))
• 그 다음 cnt.count(max(cnt))는 가장 높은 빈도수가 몇 번 등장하는 지 세어보고, 2번 이상이면 -> "?"을 출력하고, 그렇지 않으면, 'cnt.index(max(cnt))'를 통해 가장 높은 빈도수를 가진 알파벳의 인덱스를 찾고, 해당 인덱스에 해당하는 알파벳을 출력한다.

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

cnt = []
for s in alphabet:
 cnt.append(word.count(s))

if cnt.count(max(cnt)) >= 2:
 print('?')
else:
 print(alphabet[cnt.index(max(cnt))])

 

<숏코딩>

 

내 코드의 짧고 간결하게 작성되어 있는 버전이다.

word = input().upper()
a=[*{*word}]
b=[*map(word.count,a)]
m=max(b)
print([a[b.index(m)],"?"][b.count(m)>1])

 

• [*{*word}]는 'word'를 set으로 변환하여, 중복을 제거한 후 이를 list로 변환
• map(word.count, a)는 a의 각 알파벳에 대해 word.count 함수를 적용하여 빈도수를 계산하고, 이를 리스트로 변환
• b 리스트에서 가장 높은 빈도수를 max(b)로 찾음
• [a[b.index(m)], "?"][b.count(m) > 1]는 조건부 표현식으로, b.count(m) > 1이 참이면 '?'를, 거짓이면 a[b.index(m)]를 출력

 

예시