본문 바로가기

TIL/Algorithm

[프로그래머스] 큰 수 만들기 (LV2)

반응형

고려해야할 케이스는 만약 for문을 다돌고나서도 k값이 0이 아닐때, 뒤에서 k만큼 잘라줘야한다.

def solution(number, k):
    answer = ''
    stack = []
    for i in number:
        while stack and k > 0 and int(i) > stack[-1]:
            stack.pop()
            k -= 1
        stack.append(int(i))

    if k != 0:
        stack = stack[:-k]

    for i in stack:
        answer += str(i)
    return answer

 

반응형