컴퓨터는 잘못이 없다..

[알고리즘]BOJ_1157_단어 공부(파이썬 collections모듈의 Counter객체 사용, most_common함수 사용) 본문

공부/알고리즘(파이썬)

[알고리즘]BOJ_1157_단어 공부(파이썬 collections모듈의 Counter객체 사용, most_common함수 사용)

도토리까꿍v 2021. 6. 23. 15:58
Contents 접기

#문제 링크

1157번: 단어 공부 (acmicpc.net)

 

1157번: 단어 공부

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

www.acmicpc.net

 

#문제

단어 공부

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
2 초	128 MB	102687	39983	32104	38.975%
문제
알파벳 대소문자로 된 단어가 주어지면, 이 단어에서 가장 많이 사용된 알파벳이
무엇인지 알아내는 프로그램을 작성하시오. 단, 대문자와 소문자를 구분하지 않는다.

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

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

예제 입력 1
Mississipi
예제 출력 1
?
예제 입력 2
zZa
예제 출력 2
Z
예제 입력 3
z
예제 출력 3
Z
예제 입력 4
baaa
예제 출력 4
A

 

#답안

import sys,collections
input=sys.stdin.readline

#입력받은 수를 모두 대문자로 바꿔준다.
sentence=input().rstrip().upper()

#각 알파벳 개수를 세준 후, 빈도수가 가장 높은 2개를 뽑는다.
cnt=collections.Counter(sentence).most_common(2)

#위 두줄은 아래와 같이 한줄로 표현이 가능하다.
#cnt=collections.Counter(input().rstrip().upper()).most_common(2)

#두개 이상인데 두개의 빈도수가 같다면 ?를 출력한다.
if len(cnt)>=2 and cnt[0][1] == cnt[1][1] :
    print('?')
else : #한개만 출력되었거나, 2개가 출력되더라도 빈도수가 같지 않으면 맨앞에 있는 문자를 출력한다.
    print(cnt[0][0])

▲답안 설명

입력을 Mississipi로 받았다면

MISSISSIPI로 바꾼 후 단어의 빈도수를 Counter객체를 이용하여 세준다. ->  Counter({'I': 4, 'S': 4, 'M': 1, 'P': 1})

그 후 most_common(2)를 구하면 가장 빈도수가 높은 2개를 뽑아준다. -> [('I', 4), ('S', 4)] 
이때 앞과 뒤를 비교해서 빈도수가 같다면 ?를 출력하고

그게 아니라면 맨 앞의 한개만을 출력해준다. 

Comments