프로그래머스(Programmers) 큰 수 만들기 파이썬(Python) 풀이

프로그래머스(Programmers) 큰 수 만들기 lv2 파이썬(Python) 풀이

문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/42883

프로그래머스

문제 설명

양의 정수로 이루어진 배열 number에서 k 개의 수를 제거했을 때 얻는 가장 큰 수를 구하세요.

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

입출력 예

number	k	return
"1924"	2	"94"
"1231234"	3	"3234"
"4177252841"	4	"775841"

 

문제 풀이

1. 스택 이용해서 입력받은 값을 하나씩 삽입
2. 스택의 마지막 값보다 다음에 들어올 값이 작으면 스택에서 뺌
3. 반복
4. 만약 입력된 값이 정렬된 값이면 k 개만큼 뒤에서 빼주면 됨

 

소스코드

def solution(number, k):
    stack = [number[0]]
    for num in number[1:]:
        while len(stack) > 0 and stack[-1] < num and k > 0:
            k -= 1
            stack.pop()
        stack.append(num)
    if k != 0: #제대로 정렬 된 경우 다 append 됨 따라서 끝 값만 제거 하면 됨 ex 987654321 , 2 => 987654321  
        stack = stack[:-k]
    return ''.join(stack)