TIL/Algorithm
[프로그래머스] 큰 수 만들기 (LV2)
재융
2021. 1. 6. 02:43
반응형
고려해야할 케이스는 만약 for문을 다돌고나서도 k값이 0이 아닐때, 뒤에서 k만큼 잘라줘야한다.
def solution(number, k):
answer = ''
stack = []
for i in number:
while stack and k > 0 and int(i) > stack[-1]:
stack.pop()
k -= 1
stack.append(int(i))
if k != 0:
stack = stack[:-k]
for i in stack:
answer += str(i)
return answer
반응형