어떤 문자열 s가 주어졌을 때, s로부터 만들 수 있는 부분 문자열 중 사전 순으로 가장 뒤에 나오는 문자열을 찾으려 합니다. 부분 문자열을 만드는 방법은 다음과 같습니다.

s에서 일부 문자를 선택해 새로운 문자열을 만듭니다. 단, 이때 문자의 순서는 뒤바꾸지 않습니다. 예를 들어 문자열 xyb로 만들 수 있는 부분 문자열은 다음과 같습니다.

x
y
b
xy
xb
yb
xyb

이 중 사전 순으로 가장 뒤에 있는 문자열은 yb입니다.

문자열 s가 주어졌을 때 s로부터 만들 수 있는 부분 문자열 중 사전 순으로 가장 뒤에 나오는 문자열을 리턴하는 solution 함수를 완성해주세요.

제한 사항

  • s는 길이가 1 이상 1,000,000 이하인 문자열입니다.
  • s는 알파벳 소문자로만 이루어져 있습니다.

입출력 예

s result
“xyb” “yb”
“yxyc” “yyc”

나의 풀이

  • if문이 아니라 while문으로 비교해야 했다. 예를 들어 입력 값이 acbcdd일 경우 결과는 dd가 나와야하지만 cdd가 나온다. 이는 가장 뒤에 있는 값을 한 번만 비교하고 pop()을 하기 때문이다. pop()을 한 후의 또 마지막 값이 현재 값보다 작은 경우를 생각해야 한다.
def solution(s):
    stack = []
    for elem in s:
        if len(stack) > 0 and stack[-1] < elem:
            stack.pop()
        stack.append(elem)
    return "".join(stack)

다른 사람의 풀이

def solution(s):
    stack = []
    for elem in s:
        while len(stack) > 0 and stack[-1] < elem:
            stack.pop()
        stack.append(elem)
    return "".join(stack)