[백준] 1157_단어 공부 파이썬 _ set()와 count()

---

푸는데 어려움을 겪었다.

 

처음 생각했던 방법은 시간초과가 나올 것 같은 방법이었다.

배열을 만들어 a~z까지 방을 만들고, for문을 통해서 단어를 하나하나 갯수를 배열에 채우는 것이었다.

 

이 방법 말고 다음으로 택했던 방법은 dict를 사용하는 것이었다.

의도는 단어를 이루는 알파벳을 key로 하고, 갯수를 value로 해서 key와 value를 계속 update 시키는 것이었다.

그런데, 문제에 봉착한 부분이 있었는데, 그 부분은 다음의 코드를 통해 알 수 있다.

word = input()
word = word.lower()
dictionary = {}
for i in word:
    if i not in dictionary:
        dictionary = dict(i = 1)
    else:
        dictionary[i] = dictionary[i] + 1
print(max(dictionary))

set()는 중복되는 값을 없애는 것이고, count()는 set에 있는 원소들이 실제로 각각 몇 개씩 있는지 세는 것이다.

이런 함수를 잘 알고 있어야 한다는 사실을 한 번 또 느낀다.

 

따라서 다음과 같은 코드로 간단히 해결이 된다.

word = input().lower()      # word = mississipi / baaa
words = list(set(word)) # set를 통해서 중복 없애기 ==> word_list = ['m', 'i', 's', 'p'] / ['b', 'a']
cnt = []
 
for i in words:         # i = m, i, s, p / b, a
    count = word.count(i)  # count()를 통해서 set의 원소의 갯수를 센다.
    cnt.append(count)       # cnt = [4, 4, 1, 1] / [1, 3]
 
if cnt.count(max(cnt)) >= 2:
    print("?")
else:
    print(words[(cnt.index(max(cnt)))].upper())

 

문제 출처: https://www.acmicpc.net/problem/1157

1157번: 단어 공부