[python][프로그래머스] 큰수만들기

🚀  들어가며...

• LV2 문제입니다.
• 탐욕법(Greedy Algorithm)으로 접근해야 하는 문제입니다.

 

🔗 문제  

https://school.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는 2자리 이상, 1,000,000자리 이하인 숫자입니다.
• k는 1 이상 number의 자릿수 미만인 자연수입니다.

[ 입출력 예 ]

number	k	return
"1924"	2	"94"
"1231234"	3	"3234"
"4177252841"	4	"775841"

🔗 풀이

문자열로 주어진 숫자에 대해서 k개의 수를 제거하여 가장 큰 수가 되도록 만드는 문제입니다.

접근 방식만 잘 생각하면 10줄 내외로 풀이 할 수 있는 비교적 간단한 문제였지만, 알고리즘 문제 풀이를 시작한 지 얼마 되지 않아 많이 헤맸던 문제이기도 합니다.

 

# (1차풀이) 조합이용 : 시간초과 😂

시간 초과가 났던 풀이 방식입니다. 실제로도 제한 조건이 100만 자리 이하의 자연수이기 때문에 시간 초과가 날 것을 예상하긴 했지만 가장 단순한 접근이어서 풀어보았습니다.

문제의 핵심은 k개의 수를 제거한 후, 가장 큰 수를 구하는 것이기 때문에 k개를 제외한 나머지 수를 순서와 상관없이 뽑는 모든 경우의 수에서 최댓값을 구하면 된다고 생각하여 조합을 활용 후 max 함수를 이용하여 결과값을 구할 수 있었습니다.

💌 소스코드

# 1차풀이
from itertools import combinations

def solution(number, k):
    answer = list(combinations(number, len(number) - k))
    s = []
    for i in answer:
        s.append(int(''.join(i)))
    return str(max(s))

• 실행 결과
  

  

 

# (2차풀이) 스택이용: 통과 😊

이전에 풀어보았던 구명조끼 문제처럼, Stack을 이용하여 속도이슈를 해결해보자 라는 생각으로 구현했습니다.

• 핵심은 스택의 마지막 값이 push 할 값보다 작다면 크거나 같은 값이 나올 때까지 값들에 대해서 pop을 하는 것입니다.
• 이렇게 풀이하면 O(n)의 시간 복잡도로 문제를 해결할 수 있습니다.

알고리즘 

1. 스택 생성 => 파이썬에서는 리스트 활용 가능
2. number를 순회 => for i in number:
3. 조건문
   1. k > 0
   2. 스택이 비어있지 않음
   3. 스택 마지막 값 < i 
   1. 스택을 pop
   2. k--
4. 스택에 i를 push
5. k > 0 이상이면 스택에서 k개 삭제 후 join해서 결과 값 반환

풀이 예시

앞자리에 큰 숫자가 오는 것이 전체 수를 크게 만들 수 있습니다.

• number = "4177252841", k=4일 경우,
  • (k=4) []
  • (k=4) [4]
  • (k=4) [4, 1]
  • (k=3) [4]
  • (k=2) []
  • (k=2) [7]
  • (k=2) [7, 7]
  • (k=2) [7, 7, 2]
  • (k=1) [7, 7]
  • (k=1) [7, 7, 5]
  • (k=1) [7, 7, 5, 2]
  • (k=0) [7, 7, 5]
  • (k=0) [7, 7, 5, 8]
  • (k=0) [7, 7, 5, 8, 4]
  • (k=0) [7, 7, 5, 8, 4, 1]
  • retrun "775841"
• number = "999", k=2일 경우,
  • (k=2) []
  • (k=2) [9]
  • (k=2) [9, 9]
  • (k=2) [9, 9, 9]
  • return "9"

💌 소스코드

# 2차풀이
def solution(number, k):
    answer = []

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

    answer = answer[:-k] if k > 0 else answer
    return ''.join(answer)

• 실행 결과
  

  

 

# (3차풀이) 최종

1차 코드에서 중복되는 부분이나 불필요한 부분을 줄이고 최적화된 코드로 작성해보았습니다.

💌 소스코드

# 3차풀이 (2차풀이 코드정리)
def solution(number, k):
    answer = []

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

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

• 실행 결과
  

  

 

핵심 포인트

• Stack을 활용하기
• 예외 케이스까지 고려하기
• 중복된 코드 줄이고 최적화 된 코드 작성하기

 

🙋🏻‍♂️ 후기

핵심포인트가 중요한 부분이었습니다. 이 문제는 특히 예외 케이스를 놓치기 쉬운데, 평소에 연습할때는 파이참 디버거를 이용하여 어디가 잘못되었는지 확인할 수 있지만, 실제 기업의 코딩테스트를 응시할 때에는 print로 하나하나 찍어보면서 해야하기 때문에 시간이 많이 소요될것 같습니다. 평소에도 이런 연습을 많이 해둬야 겠습니다.