프로그래머스: 큰 수 만들기

문제

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

 

문제 풀이

 어떤 숫자에서 K개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하는 문제이다. 문제를 풀기 위해서는 stack을 사용하면 해결할 수 있다.

 

• stack에 값이 없다면 수를 append 한다.
• stack에 값이 있다면 현재의 수와 top을 비교하여 값이 값이 크지 않는 경우 pop한다.
  • 문제에 제거할 수 있는 K가 주어지므로, 이 역시 같이 판단하여야 한다.
• 만약 K가 0이 된다면 즉시 중단하고, 남은 수들을 stack에 삽입한다.

 

코드

def solution(number, k):
    stack = []

    for i, num in enumerate(number):
        while stack and stack[-1] < num and k > 0:
            stack.pop()
            k -= 1

        if k == 0:
            stack += number[i:]
            break

        stack.append(num)

    stack = stack[:-k] if k > 0 else stack
    return "".join(stack)