코테 연습> 탐욕법(Greedy) 큰 수 만들기

Doyeon Kim·2022년 1월 11일

코딩테스트 공부

목록 보기
10/171

문제 설명
어떤 숫자에서 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 k return
"1924" 2 "94"
"1231234" 3 "3234"
"4177252841" 4 "775841"


간단하게 Number에 있는 수들 중에 k개의 수만큼 제거하고 난 뒤 만들 수 있는 수 중 가장 큰 숫자를 만들라고하는 것은 이해가 갔다.
문제를 파악하는 것은 어렵지는 않았는데.. 뭔가 테스트케이스에 통과하도록(?) 실제 코드로 구현하는 것이 생각보다 쉽지는 않았다.


def solution(number, k):
    answer = []

    for i in number:
        while len(answer)>0 and k>0 and answer[-1]<i:
            answer.pop()
            k-= 1
        answer.append(i)

    answer =''.join(answer[:len(number)-k])

    return answer

먼저 answer 배열을 만든 뒤에
answer에 값이 존재하고, k가 0보다 크고 answer의 이전 값이 number의 현재 값보다 작다면 기존 값을 pop하고 k의 값을 1 줄이고 새로운 값을 append한다.

그리고 예를 들어 ex) k = 3 number = 1000000 이런 경우엔 k는 처음 인풋받은 그대로 유지된다

# 이럴 때 답은 뒷 숫자를 k개만큼 없애준 1000 이므로 슬라이싱을 len(number) - k로 해주는 것이다

2022.01.31  복습
마지막 answer =''.join(answer[:len(number)-k])예를 들어  ex) k = 3 number = 1000000 이런 경우엔 k는 처음 인풋받은 그대로 유지된다
# 이럴 때 답은 뒷 숫자를 k개만큼 없애준 1000 이므로 슬라이싱을 len(number) - k로 해주는 것이다
이부분 어려웠음
profile
성장하고 도전하는 개발자. 프로그래밍 좋아하세요?

0개의 댓글