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

LV.2 큰 수 만들기

문제 설명

어떤 숫자에서 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"

---

문제 요약: 숫자 number에서 k개 만큼의 수를 제거하고 만들 수 있는 수 중 가장 큰 수 찾기

생각할 점:

• 크기를 비교하는 숫자들의 자릿수는 모두 같다.
  • 정답 문자열 answer의 길이는 number.size() - k
• 숫자 문자열의 첫 인덱스가 높은 숫자일 수록 크다
• 인덱스 순으로 큰 숫자인지를 비교
• 그리디 알고리즘!

#include <string>
#include <vector>
 
using namespace std;
 
string solution(string number, int k) {
    string answer = "";
 
    int numSize = number.size() - k;
    int index = 0;
    
    for(int i=0; i < numSize; i++) {
        char maxNum = number[index];
        int maxIdx = index;
        for(int j = index; j <= k+i; j++) {
            if(maxNum < number[j]) {
                maxNum = number[j];
                maxIdx = j;
            }
        }
        index = maxIdx + 1;
        answer += maxNum;
    }
 
    return answer;
}

주석 추가 버전

#include <string>
#include <vector>
 
using namespace std;
 
string solution(string number, int k) {
    string answer = "";
 //정답 배열의 길이는 number.size() -k
    int numSize = number.size() - k;
 //탐색을 시작할 인덱스
    int index = 0;
    //인덱스 0번 부터 i + numsize 만큼 반복
    for(int i=0; i < numSize; i++) {
			//number에서 가장 큰 수의 값을 maxNum에 저장
        char maxNum = number[index];
			//가장 큰 수의 인덱스를 maxIdx에 저장
        int maxIdx = index;
			//정답이 될 수의 배열에서 가장 큰 수를 찾아서 갱신해주기
        for(int j = index; j <= k+i; j++) {
            if(maxNum < number[j]) {
                maxNum = number[j];
                maxIdx = j;
            }
        }

        index = maxIdx + 1;
        answer += maxNum;
    }
 
    return answer;
}