728x90

문제 설명

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

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

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

제한 조건

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

입출력 예

number k return

1924 2 94
1231234 3 3234
4177252841 4 775841

그리디 섹션에 있는 문제이다.

문자열에서 입력으로 들어온 k 만큼의 수를 제거해서 가장 큰 수를 만들어야 한다.

 

해결 방법은 문자열의 제일 앞에서부터 고를 수 있는 가장 큰 수를 고르는 것이다.

{1231234, 3}을 예로들면 정답 문자열의 길이는 4가 될 것이다.

먼저 제일 앞에서부터 생각하면, 1231중에서 가장 큰 값을 하나 골라야 한다. 왜냐하면 그래야 정답 문자열의 길이보다 짧아지지 않기 때문이다. 만약 제일 처음에 4를 뽑으면 그 후에는 고를 수가 없어진다. 

아무튼, 1231중에 가장 큰 수는 3이기 때문에 3을 고르고, 그 다음엔 12중에 가장 큰 수인 2를 고르고 3, 4를 고르면 정답 문자열 3234가 나오게 된다.

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/Programmers/%ED%81%B0%20%EC%88%98%20%EB%A7%8C%EB%93%A4%EA%B8%B0.cpp

 

chosh95/STUDY

프로그래밍 문제 및 알고리즘 정리. Contribute to chosh95/STUDY development by creating an account on GitHub.

github.com

 

 

C++ 코드

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

string solution(string number, int k) {
	string answer = "";
	int cur = 0;
	for (int i = 0; i < number.size()-k; i++) {
		char maxNum = number[cur];
		int maxIdx = cur;
		for (int j = cur; j <= k + i; j++) {
			if (maxNum < number[j]) {
				maxNum = number[j];
				maxIdx = j;
			}
		}
		cur = maxIdx + 1;
		answer += number[maxIdx];
	}
	return answer;
}

int main()
{
	cout << solution("4177252841", 4);
}
728x90

+ Recent posts