URL : https://programmers.co.kr/learn/courses/30/lessons/42883
문제 설명
- 어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.
- 예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.
- 문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.
제한 조건
- number는 1자리 이상, 1,000,000자리 이하인 숫자입니다.
- k는 1 이상 number의 자릿수 미만인 자연수입니다.
문제 풀이
- 완전 탐색으로하기엔 시간이 오래걸릴것이 이미 문제에 적혀있었다
- 그래서 number의 원소를 하나씩 리스트에 넣고, 그 뒤에 원소와 비교하여 작으면 해당 원소를 지우는 방식으로 풀었음
- 일단 빈 리스트를 만들고, 그 뒤에 number의 원소를 하나씩 넣음
- 하나씩 넣은 원소가 뒤에것보다 작으면 해당 원소를 지우고 리스트에 뒤에 온 원소를 넣고 k를 1씩 줄여나가면 k가 0이 될떄는 해당 while문이 끝남
- 하지만 문제는 처음부터 while문이 False여서 실행되지 않는것. 그렇다면 ls = number가 될것
- 그래서 만약 k가 0보다 크다면 ls = ls[:-k]의 if문을 추가하여 주었음
- ls[:-k] 는 맨뒤에 k를 뺀 리스트라는 의미
1
2
3
4
5
6
7
8
9
10
11
def solution(number, k):
ls = []
for i, num in enumerate(number):
while len(ls) > 0 and ls[-1] < num and k > 0:
ls.pop()
k -=1
ls.append(num)
ls = ls[:-k] if k > 0 else ls
return ''.join(ls)
1
2
3
number = '99991'
k = 3
solution(number,k)
1
'99'
1
2
3
number = '111119'
k = 3
solution(number,k)
1
'119'
1
2
number = '123456'
ls = [i for i in number]