프로그래머스 큰수만들기 - peulogeulaemeoseu keunsumandeulgi

문제 설명

어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.

예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.

문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.

제한 조건

• number는 1자리 이상, 1,000,000자리 이하인 숫자입니다.
• k는 1 이상 number의 자릿수 미만인 자연수입니다.

입출력 예

풀이 코드 [자바/java]

class Solution {
    public String solution(String number, int k) {
        StringBuilder sb = new StringBuilder();
        int index = 0;
        int max = 0;
        for(int i=0; i<number.length() - k; i++) {
            max = 0;
            for(int j = index; j<= k+i; j++) {
                if(max < number.charAt(j)-'0') {
                    max = number.charAt(j)-'0';
                    index = j+1;
                }
            }
            sb.append(max);
        }
        return sb.toString();
    }
}

내 문제풀이 방법 설명

먼저 return 해야 할 문자의 길이는 매개변수로 들어온 number의 길이에서 k만큼을 빼준 길이가 된다.

그래서 for문을 먼저 만들어야 할 문자의 길이만큼 반복을 했다.

그 이후에 가장 큰 수를 탐색하여 StringBuilder에 붙혀주고 return 해주면 되는데..

조건에서 k만큼의 수를 제거했을때 가장 큰 수 이기 때문에

두번째 for문의 범위는 탐색해야하는 문자의 시작(index)부터, 앞으로 이어붙혀야할 문자의 길이 -1이 뒤에 남는 index까지 탐색을 해야한다.

그리고 index에는 가장 큰수 다음 index를 넣어준다.

이유는 그 다음 문자열부터 가장 큰 수를 찾아야하기 때문이다.

[입출력 예]의 3번 예제로 예를 들면

'4177252841' 를 0번 index부터 탐색을 시작하여 총 6자리의 return 값을 만들어야 하기 때문에 뒤에서 6번째에 위치한 '41772' 까지 탐색을 진행한다.

그리고 StringBuilder에 가장 큰 수를 append한다.

그러면 이제 앞으로 5자리를 이어 붙히면 되고, 이제 탐색할 시작 index는 가장 큰 수였던 3번인덱스 다음인 4번인덱스 부터 순회한다.

'7252841' 문자열을 탐색하는것이다.

그러면 앞의 규칙에 따라 뒤에서 5번째에 위치한 '725' 까지 탐색을 진행하고 가장큰수를 append한다.

그 다음은 '252841' 문자열을 뒤에서 4번째인 '252' 까지,

그 다음은 '2841' 문자열을 뒤에서 3번째인 '28' 까지,

그 다음은 '41' 문자열을 뒤에서 2번째인 '4' 까지,

그 다음은 '1' 문자열을 뒤에서 1번째인 '1' 까지,

모든 범위가 끝나게되면 현재까지 진행했던 가장큰수를 이어붙인 '775841'가 만들어질것이고

return 해주면 된다.

🚀  들어가며...

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

🔗 문제  

https://school.programmers.co.kr/learn/courses/30/lessons/42883

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

📑 내용

[ 문제 설명 ]

어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.

예를 들어, 숫자 1924에서 수 두 개를 제거하면 [19, 12, 14, 92, 94, 24] 를 만들 수 있습니다. 이 중 가장 큰 숫자는 94 입니다.

문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어집니다. number에서 k 개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하도록 solution 함수를 완성하세요.

[ 제한 조건 ]

• number는 2자리 이상, 1,000,000자리 이하인 숫자입니다.
• k는 1 이상 number의 자릿수 미만인 자연수입니다.

[ 입출력 예 ]

🔗 풀이

문자열로 주어진 숫자에 대해서 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로 하나하나 찍어보면서 해야하기 때문에 시간이 많이 소요될것 같습니다. 평소에도 이런 연습을 많이 해둬야 겠습니다.