[BAEKJOON] 1157 단어 공부

문제

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

입력

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

출력

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

예제 입력 1 복사

Mississipi

예제 출력 1 복사

?

예제 입력 2 복사

zZa

예제 출력 2 복사

Z

예제 입력 3 복사

z

예제 출력 3 복사

Z

예제 입력 4 복사

baaa

예제 출력 4 복사

A

---

이 문제 같은 경우는 count를 할 수 있는가에 대해 문제였다.

내 습관인지 모르겟지만 이런 갯수를 세는 문제에서는 딕셔너리 접근이 제일 먼저 생각났다.

 

def solution(w):

    ret = {}

    w = w.lower()

    for c in w:

        cnt = w.count(c)

        if(not ret.get(cnt)):

            ret[cnt] = [c]

        else:

            if(not c in ret[cnt]):

                ret[cnt].append(c)

    ret = dict(sorted(ret.items(), key=lambda x:x[0], reverse=True))

   

    v = ret[list(ret.keys())[0]]

    if(len(v) > 1):

        return '?'

    else:

        return v[0].upper()

   

a = frs

print(solution(a))

그래서 처음 접근한 방법

근데 시간 초과가 떳다...

 

그래서 생각을 해보니 단어의 길이가 1,000,000을 넘지 않는 다는 것이다.

그렇다면 지금 내 코드는 1,000,000 번의 반복이 일어나고 있다는 것...?

 

그래서 간소화를 할 필요가 있었다.

 

그래서 set 함수가 생각났다.

 

import sys

import copy

import math

fr = sys.stdin.readline()

frs = fr.strip()

frl = fr.split()

def solution(w):

    w = w.upper()

    ret = list(set(w))

    rcnt = []

    for c in ret:

       cnt = w.count(c)

       rcnt.append(cnt)

   

    if rcnt.count(max(rcnt)) > 1:

        return "?"

    else:

        return ret[rcnt.index(max(rcnt))]

 

 

a = frs

print(solution(a))

그래서 두번 째 작성한 코드