[BAEKJOON] 1157번 : 단어 공부

파이썬으로 알고리즘을 풀어보자.

파이썬이랑 친해지기위해 브론즈 문제부터 차근차근 풀어보려고한다.

 

다음 문제를 풀면서 파이썬의 dictionary를 학습했다.

 

문제

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

입력

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

출력

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

파이썬이랑 친해지기

dictionary 사용하기

python의 dictionary는 key, value 형태의 자료형으로 java의 Map과 유사한 형태를 띄고있다.

python의 dictionary는 순서가 없다.

 

다음과 같이 선언하고 값을 변경할 수 있다.

dic = {'a': 1, 'b': 2}

dic['a'] = 4

 

max 함수사용하기

python의 max, min 함수는 iteratable한 자료형을 넣으면 해당 값에서 최대, 최소값을 가져온다.

 

dic = { 'a':1, 'b':2 }
max(

 

두 함수의 콜라보

두 함수를 적절히 섞어서 사용하여 가장 많이 사용된 알파벳을 뽑아보자.

max 함수를 사용할 때 key라는 인자를 사용하면 어떤 값을 기준으로 최댓값을 구할 지 지정할 수 있다.

아래 코드를 해석하면 dictionary에서 get으로 얻어오는 value를 기준으로 최댓값을 구한다. 라고 볼 수 있다.

str = input().upper()

dic = {}

for i in str:
    if i not in dic:
        dic[i] = 0
    dic[i] += 1

print(max(dic, key=dic.get))

이제 문제 조건을 만족시키기 위해 두개 이상의 최댓값이 존재하면 ?를 띄우기만 하면 된다.

코드

str = input().upper()

dic = {}

for i in str:
    if i not in dic:
        dic[i] = 0
    dic[i] += 1

result = []
m = max(dic.values())
for key, val in dic.items():
    if m == val:
        result.append(key)

if len(result) != 1:
    print('?')
else:
    print(result[0])

 

 

 

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

1157번: 단어 공부