[백준] 21314번 민겸 수 (python)

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

21314번: 민겸 수

 

가장 큰 수를 구하는 방법

• 문자열을 하나씩 탐색하다가 K를 만나면 끊어준다.
• 만약 문자열이 K로 끝나지 않으면? K로 끊은 이후의 문자열은 하나의 M으로 취급해 준다. (예를 들어 KMKMM인 경우 K, MK, M, M)

 

가장 작은 수를 구하는 법

• 문자열을 하나씩 끊어준다.
• 단, M이 연속되는 경우는 묶어준다. (예를 들어 MKKMMK인 경우 M, K, K, MM, K)

 

정답 코드

import sys
input = sys.stdin.readline

strings = list(input().strip())

# 가장 큰 수를 구하는 로직
temp = ""
before = 0
big_values = []
for index, string in enumerate(strings):
    temp += string

    if string == "K":
        big_values.append(temp)
        temp = ""
    
    if index == len(strings) - 1 and temp != "":
        for _ in range(len(temp)):
            big_values.append("M")

answer = ""
for value in big_values:
    if "K" in value:
        answer += str(5 * (10 ** value.count("M")))
    else:
        answer += str(10 ** (value.count("M")-1))
print(answer)


# 가장 작은 수를 구하는 로직
temp = ""
before = 0
small_values = []
for index, string in enumerate(strings):
    if string == "M":
        temp += string

    if string == "K":
        small_values.append(temp)
        small_values.append("K")
        temp = ""

    if index == len(strings) - 1 and temp != "":
        small_values.append(temp)

small_values = [value for value in small_values if value != ""]
answer = ""

for value in small_values:
    if "K" in value:
        answer += str(5 * (10 ** value.count("M")))
    else:
        answer += str(10 ** (value.count("M")-1))
print(answer)