[Python][백준] 1157. 단어 공부

링크🔗

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

---

🗒️파이썬  풀이

str = input().upper()
lst_str = list(set(str))
lst = []

for word in lst_str:
    lst.append(str.count(word))


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

 

1. 대소문자 구분을 안하고, 대문자로 출력하기 떄문에 upper 함수 사용

 

2. 각 문자의 개수를 파악하기 위해 집합으로 만든 후 리스트 생성

 

3. 집합으로 만들어진 리스트의 단어들을 통해 str 단어 개수 파악 후 리스트 append

 

4. 각 단어의 개수가 들어가있는 lst의 max 값이 2개 이상인 경우 "?" 출력

그게 아닌 경우에는, lst에서 가장 큰 값의 인덱스 추출 후  집합 리스트 인덱스에 부여

 

🔎 내 풀이

str = input()
lst = [str[i].upper() for i in range(len(str))]


set_lst = set(lst)

cnt_lst = []

for word in set_lst:
    cnt_lst.append(lst.count(word))

mx = max(cnt_lst)

cnt = 0 
for word in set_lst:
    if mx == lst.count(word) : 
        cnt += 1

if cnt > 1: 
    print("?")
else:
    for word in set_lst:
        if mx == lst.count(word) : 
            print(word)

 

내가 작성한 코드에는 불필요한 중복들이 너무 많다.  

 

1. set_lst로 만들고 count 함수로 mx 값 만듬

 

2. mx와 set_lst로 최대값이 2개 넘는지 확인

 

3. 1개 이상일때 조건, 1개일때 조건으로 나눔

(1개일때에 조건에 맞는 단어 추출)

 

기본적인 로직은 같지만, 이것들을 잘 줄여서 시간 복잡도와 공간 복잡도를 줄여보자. 

 

📌  문제 코멘트

upper, count, index, max 기본 내장 함수를 알면 쉽게 풀 수 있었다.

다만 set 같은 경우 인덱싱이 안되므로 list로 변환하여 사용하는게 좋다.

 

---

📚문제

 

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

입력

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

출력

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

예제 입력 1 복사

Mississipi

예제 출력 1 복사

?

예제 입력 2 복사

zZa

예제 출력 2 복사

Z

예제 입력 3 복사

z

예제 출력 3 복사

Z

예제 입력 4 복사

baaa

예제 출력 4 복사

A