게임 클라이언트 개발/알고리즘 문제
[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;
}