반응형
고려해야할 케이스는 만약 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
반응형
'TIL > Algorithm' 카테고리의 다른 글
[백준] 코딩연습 (0) | 2023.05.16 |
---|---|
[HackerRank] Subarray Division (0) | 2021.02.22 |
[프로그래머스] 키패드 누르기(easy, kakao) (0) | 2021.01.04 |
[프로그래머스] 크레인 인형뽑기 (easy, kakao) (0) | 2021.01.04 |
[프로그래머스] 실패율(easy, kakao) (0) | 2021.01.04 |