[Programmers][python] 23. 문제풀이 실습 (11): 프로그래머스 큰 수 만들기

오류에 대한 지적이나 질문, 토의 환영합니다. 자유롭게 댓글 남겨주세요!.!

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

프로그래머스 큰 수 만들기 문제의 링크입니다.

문제 설명

어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.

예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.

문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.

제한사항

• number는 1자리 이상, 1,000,000자리 이하인 숫자입니다.
• k는 1 이상 number의 자릿수 미만인 자연수입니다.

입출력 예

생각

1. 가장 큰 수를 만들기 위해 가장 쉬운 방법은 모든 조합을 고려해보는 것일 것입니다. 그러나, number의 수를 봤을 때 최대 백 만 자리인 굉장히 큰 입력이 들어올 수 있으므로 $O(n^2)$ 보다 빠른 복잡도를 가진 알고리즘으로 풀어야 합니다.

2. • k개의 숫자를 제거할 때, 앞에서부터 작은 숫자 k 개를 제거하는 탐욕법의 방법을 선택하고 싶습니다.
   • 앞 단계에서의 선택이 이후 단계에서의 최적의 선택을 해치지 않으므로(독립적이므로) 탐욕법이 통할 것입니다.
   • number의 자리 수 만큼 반복하면 될 것이므로 $O(n)$의 복잡도를 가집니다.

3. 수를 선택했을 때,

   • 스택이 비어있으면 수를 스택에 넣고,
   • 스택이 비어있지 않다면,
     • 스택의 top에 들어있는 수가 선택된 수보다 크다면 선택된 수를 스택에 넣습니다.
     • 스택의 top에 들어있는 수가 선택된 수보다 작다면 스택의 top에 있는 수를 pop하고 선택된 수를 넣습니다.

4. 이것을 k개만큼 반복하고 k만큼 반복되지 않았다면 나머지만큼 뒤에서 잘라줍니다.
   

Code (Python)

def solution(number, k):
    answer = []
    
    for num in number:
        while len(answer) != 0 and answer[-1] < num and k > 0:
            answer.pop()
            k -= 1
        answer.append(num)
        
    answer = answer[:-k] if k > 0 else answer
    
    return ''.join(answer)

이 글은 프로그래머스 스쿨 인공지능 데브코스 과정에서 공부한 내용을 바탕으로 정리한 글입니다.