[프로그래머스] 큰 수 만들기 - Lv.2

프로그래머스 42883번 문제 "큰 수 만들기" Lv.2 - 파이썬, 자바

[프로그래머스] 큰 수 만들기 Lv.2 - [파이썬/python] [자바/java]

 

 

---

💻 문제 설명

 

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

 

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

 

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

 

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

프로그래머스

---

🚨 제한사항

 

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

---

 

 

---

---

---

!!!정답 주의!!!

---

---

---

 

 

 

---

🌟 소스 코드

[Python] 파이썬 코드

def solution(number, k):

   

    stack = []

   

    # 모든 원소를 stack에 넣어가면서

    # 현재 넣는 원소보다 stack의 맨 위 원소가 작을 경우

    # stack의 맨 위 원소 제거

    for num in number:

       

        while stack and stack[-1] < num and k > 0:

            stack.pop()

            k -= 1

           

        stack.append(num)

   

    # k가 남을 수도 있음

    # 뒷 원소 k만큼 짜르기

    stack = stack[:len(number)-k]

   

    # 문자열로 반환

    return ''.join(stack)

 

 

 

[Java] 자바 코드

import java.util.Stack;

class Solution {
    public String solution(String number, int k) {
        
        char[] result = new char[number.length() - k];
        Stack<Character> stack = new Stack<>();

        // 모든 원소를 stack에 넣어가면서
        // 현재 넣는 원소보다 stack의 맨 위 원소가 작을 경우 stack의 맨 위 원소 제거
        for(char num : number.toCharArray()){

            while (!stack.empty() && stack.peek() < num && k > 0){
                stack.pop();
                k--;
            }

            stack.add(num);
        }
        
        // stack의 원소를 result에 옮기고
        for(int i=0; i<result.length; i++){
            result[i] = stack.get(i);
        }
        
        // String으로 변환 후 반환
        return new String(result);
    }
}

 

 

 

---

🏳‍🌈 테스트 결과

 

 

[Python] 파이썬 - 테스트 결과	[Java] 자바 - 테스트 결과