[프로그래머스] 큰 수 만들기 .py

programmers.co.kr/learn/courses/30/lessons/42883

코딩테스트 연습 - 큰 수 만들기

def solution(number, k):
    n = len(number)
    answer = [number[0]]
    
    for num in number[1: ]:
        while answer and answer[-1] < num and k > 0: # answer 마지막 숫자가 num보다 작으면 안됨
            answer.pop()
            k -= 1 # if k == 0 이면 주어진 k를 다 쓴 것
        answer.append(num)
        
    if k != 0: # 만약 k를 다 쓰지 않았다면
        answer = number[: -k]
        
    return ''.join(answer)

 

solution("9999999", 2)인 경우에 조심해주자