12-kokeunho

[kokeunho]

🔗 문제 링크

[BOJ] 크게 만들기 https://www.acmicpc.net/problem/2812

✔️ 소요된 시간

50min

✨ 수도 코드

높은 자리 숫자가 커질 수록 큰 수가 됩니다.
그러므로 입력 받은 숫자 문자열을 왼쪽에서 오른쪽으로 이동하면서
대소를 비교하여 큰 숫자를 남깁니다.
이러한 동작을 숫자 제거 횟수를 다 사용하거나
스택이 빌 때까지 반복합니다.

 while(!stack.isEmpty() && count < k && stack.peek() < c) {
                stack.pop();
                count++;
 }

📚 새롭게 알게된 내용

풀이는 쉽게 생각해냈는데
스택 활용은 생각하지 못하고 시간만 버리다 검색 찬스 사용했습니다.
자료구조 공부를 좀 해야겠습니다...
크리스마스 잘 보내세요

[kangrae-jo]

이 문제도 자료구조 시간에 했었던 것 같네요

전체 구조는 근호님 코드와 비슷하긴한데 while을 구현한 부분이 조금 다릅니다.
저는 count++하며 조건을 검사한게 아니라 K-- 하며 조건을 검사했습니다.

제가 한 방식에서는 살짝 문제가 있는게...
조건이 잘 맞는다면 K개 숫자를 모두 제거하지 못한채로 for문이 끝날 수 가 있었습니다.
( 참고 자료 )

그래서 for문이 끝나고도 남아있는 K를 제거해주는 로직을 적어두어야했습니다.

그리고 출력할 때 stack을 뒤집어야하는데... Java에서는 stack도 인덱스 접근이 되나보군요
c++에서는 인덱스 접근이 불가능해서 top부터 empty까지 result에 저장 후, result를 뒤집고 출력하는....
그런 로직밖에 생각이 안나더라구요

(그리고 앞에서 말한 K를 제거해주는 로직도, stack에서 인덱스 접근이되면 근호님 처럼 n - k 을 하는 것이 아주 좋을 것 같아요)

CPP CODE

#include <iostream>
#include <stack>
#include <string>
#include <algorithm>

using namespace std;

int main() {
    int N, K;
    string num;

    cin >> N >> K;
    cin >> num;

    stack<int> s;
    for (int i = 0; i < N; i++) {
        int d = num[i] - '0';

        while (!s.empty() && d > s.top() && K > 0) {
            s.pop();
            K--;
        }
        s.push(d);
    }

    // 남아있는 K만큼 제거
    while (K > 0) {
        s.pop();
        K--;
    }

    // 스택을 뒤집어서 출력
    string result;
    while (!s.empty()) {
        result += (s.top() + '0');
        s.pop();
    }
    reverse(result.begin(), result.end());

    cout << result;

    return 0;
}

(TMI: 예전에 제가 푼 것을 찾아봤는데 c로 작성해서 직접 stack을 구현하고 인덱스 접근을 했더라구요.)

TMI CODE

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <string.h>

#define MAX 1000000

int d[MAX];
int top = -1;

void push(int num) {
	top++;
	d[top] = num;
}

void pop() {
	top--;
}

int main() {
	char n[MAX];
	scanf("%s", n);

	int k;
	scanf("%d", &k);
	int k_ = k;

	int len = strlen(n);
	for (int i = 0; i < len; i++) {
		char digit = n[i];
		int num = digit - '0';

		while (top >= 0 && num > d[top] && k > 0) {
			pop();
			k--;
		}
		push(num);
	}
	
	for (int i = 0; i < len-k_; i++)
		printf("%d", d[i]);

	return 0;
}

[wnsmir]

처음에는 앞의수가 뒤의 수보다 작으면 삭제하고 이를 1의자리까지 반복하되, 삭제횟수가 K를 넘으면 종료, 끝까지 가도 못채울경우 제일 뒤에서 부터 삭제 라고 생각하고 풀었습니다.

몇가지 테스트를 해보니 바로 잘못됨을 알았습니다.
41712의 경우 1이 7보다 작아 지우게되면 4712가 되고 그대로 넘가면 1이 2보다 작으니 472가 되는 상황이 발생했습니다.

즉 현재수가 다음수보다 크면 다음수를 삭제하고, 다시 처음으로 돌아가 4와 7을 비교해서 4를 지우는 방식을 채택해야, 앞자리수가 가장 큰 수가 되는 것입니다.

이방식에 효율적인 자료구조는 리스트보단 스택이라 생각되어 스택을 활용하여 문제를 풀었습니다.

def remove_digits(number, K):
    num_str = list(str(number))  # 숫자를 문자열로 변환하여 리스트로 만듦
    stack = []  # 스택을 사용하여 숫자 비교 및 삭제 작업 진행

    for digit in num_str:
        # 스택의 마지막 숫자와 현재 숫자를 비교하여 지울지 결정
        while stack and K > 0 and stack[-1] < digit:
            stack.pop()  # 작은 숫자는 삭제
            K -= 1  # 삭제 횟수 1번 감소
        stack.append(digit)  # 현재 숫자는 스택에 추가

    # K번 삭제가 끝나지 않았으면 뒤에서부터 삭제
    while K > 0:
        stack.pop()
        K -= 1

    # 스택에 남아있는 숫자를 하나로 합쳐서 반환
    return ''.join(stack)

N, K = int(input().split())
number = list(map(int, input()))
print(remove_digits(number, K)

[g0rnn]

예전에 풀어봤던 문제라 반가웠습니다! 근데 풀이가 기억이 안나서 오랫동안 풀었습니다ㅜㅜ

저도 강래님처럼 cpp에서 stack을 사용해서 답을 출력할 때 어떻게 해야할지 고민했었는데 그냥 deque로 바꾸어서 풀었습니다.

리뷰를 적을 때 다시 생각해보니 그냥 vector를 사용하여 push_back과 pop_back으로 stack처럼 사용하면 됐었네요..;

for (int i = 0; i < n; i++) {
    int num = raw[i] - '0';
    while (k > 0 && !dq.empty() && dq.back() < num) {
        dq.pop_back();
        k--;
    }
    dq.push_back(num);
}

// k가 남았으면 뒤의 숫자를 없앰 ex) input: 54321, k: 2 -> ans: 543
while (k-- > 0) dq.pop_back();
while (!dq.empty()) {
    cout << dq.front();
    dq.pop_front();
}

[g0rnn]

위의 for문을 보고 내림차순으로 정렬된 수는 올바르게 제거하지 못하지 않나? 라고 생각했었는데 n-k까지만 출력하여 자연스레 제거하셨네요! 저는 반복문 한번 더 썼는데 이 방법이 더 효율적인것 같습니다. 👍 👍