어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.
예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.
문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.
def solution(number, k):
answer = []
for n in number: # 숫자를 순차적으로 하나씩 list에 넣기위한 반복
while answer and k > 0: # list에 내용물이 있을 때, k개 이하의 숫자를 제외했을 때 반복
if n > answer[-1]: # 넣으려는 수보다 작은 list의 수를 하나씩 pop
answer.pop()
k -= 1 # 숫자 하나 제외되면 k-1
else: # 문제 없으면 pop 반복 종료
break
if len(answer) < len(number) - k: # 필요한 자리수를 넘어가면 추가하지 않음
answer.append(n)
return "".join(answer)
예 : "4321", 1
) 정답을 저장하는 list 의 길이를 비교하여 숫자를 추가해야만 한다. "1" < "2"
와 같이 str 형태의 숫자도 대소 비교가 가능하다.