게임 클라이언트 개발/알고리즘 문제

[2] 큰 수 만들기 (그리디, 순회 stack 이용)

주운녕 2025. 2. 2. 15:18

문제 설명
어떤 숫자에서 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"


문제 풀이

순회하면서 stack에 숫자를 넣고,
현재 stack 보다 순회하는게 더 크면 빼기

뒤에 있는 큰 수 찾기 <- 이 문제와 비슷한 유형

#include <string>
#include <vector>
#include <algorithm>
#include <stack>

using namespace std;

string solution(string number, int k) {
	string answer = "";

	stack<string> s;
	int minusCnt = 0;
	for (int i = 0; i < number.length(); i++)
	{
		string c;
		c += number[i];
		while (s.empty() == false && minusCnt < k)
		{
			string t = s.top();
			if (stoi(t) < stoi(c))
			{
				s.pop();
				minusCnt++;
			}
			else
				break;
		}
		s.push(c);
	}
	while (s.empty() == false)
	{
		answer += s.top();
		s.pop();
	}
	reverse(answer.begin(), answer.end());
	answer = answer.substr(0, answer.size() - (k - minusCnt));
	return answer;
}