코테
[프로그래머스] 큰 수 만들기
코딩마스터!
2021. 9. 21. 00:12
어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.
예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.
문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.
제한 조건
- number는 1자리 이상, 1,000,000자리 이하인 숫자입니다.
- k는 1 이상 number의 자릿수 미만인 자연수입니다.
입출력 예
numberkreturn
"1924" | 2 | "94" |
"1231234" | 3 | "3234" |
"4177252841" | 4 | "775841" |
- while stack 이라는 조건이 stack 이 비어있지 않는 한 참이라는 조건인 것을 처음 알았다.
stack에 차례대로 넣어주고 자기앞의 수가 자기보다 작으면 pop 해준다
def solution(number, k):
answer = ''
number = list(map(int,str(number)))
#number 수를 하나씩 문자로 받아서 int로 변화해서 저장
stack = []
for x in number:
while stack and k>0 and stack[-1]<x:
stack.pop()
k-=1
stack.append(x)
if k != 0:
stack = stack[:-k]
answer = "".join(map(str,stack))
return answer