[C++/PGS] Lv.2 : 큰 수 만들기 (GREEDY)

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의 자릿수 미만인 자연수입니다.

---

알맞은 알고리즘을 떠올리기 힘들었다.

그리디니까 그냥 앞뒤만 비교하면 되려나?!하고 처음에는 숫자의 앞/뒤만 비교해서 더 큰수를 남겨놓고,

작은 수를 erase하는 방법으로 하였는데

어찌저찌 코드실행 테케는 다 맞았지만, 채점의 결과는 처참했다....

 

그래서 뭔가 이 방법이 아닌듯해서 구글링했음. 🤓

설명을 읽고도 이해가 바로 안되어서 여러번 생각해보고 정리해보았다.

 

 

1. number과 answer은 앞에서부터 순서가 그대로 이어지고, number.size() - k 개의 숫자가 필요하므로

    i번째로 number의 숫자를 뽑으려면 그 뒤에 최소한 number.size()-k-i 개의 숫자가 더 있어야 한다.

 

2. number.size() - k - i = number.size() - ( k + i )

 

3. 따라서 number인덱스 탐색 시 뽑을 숫자의 인덱스 영역은 start ~ k+i 까지이다.

 

 

 

 

나의 풀이

#include <string>
#include <vector>
using namespace std;

string solution(string number, int k) {
    string answer = "";
    int answersize = number.size() - k;
    int idx = 0;
    for(int i = 0; i < answersize; ++i){
        
        char maxint = number[idx];
        int maxidx = idx;
        
        for(int j = idx; j <= k+i; j++){ // k + answersize = number.size()
            if(maxint < number[j]){
                maxint = number[j];
                maxidx = j;
            }
        }
        idx = maxidx + 1;
        answer += maxint;
    }
    
    return answer;
}