본문 바로가기
Algorythms

백준 9935번 문자열 폭발 c++ 풀이

by 준형코딩 2024. 3. 22.

문제

상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.

폭발은 다음과 같은 과정으로 진행된다.

  • 문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
  • 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
  • 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.

상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.

폭발 문자열은 같은 문자를 두 개 이상 포함하지 않는다.

입력

첫째 줄에 문자열이 주어진다. 문자열의 길이는 1보다 크거나 같고, 1,000,000보다 작거나 같다.

둘째 줄에 폭발 문자열이 주어진다. 길이는 1보다 크거나 같고, 36보다 작거나 같다.

두 문자열은 모두 알파벳 소문자와 대문자, 숫자 0, 1, ..., 9로만 이루어져 있다.

출력

첫째 줄에 모든 폭발이 끝난 후 남은 문자열을 출력한다.

예제 입력 1 복사

mirkovC4nizCC44
C4

예제 출력 1 복사

mirkovniz

예제 입력 2 복사

12ab112ab2ab
12ab

예제 출력 2 복사

FRULA

 

 

문제 풀이를 이해하게 되면 구현은 쉬웠던 문제

스택에 문자를 넣고 폭탄의 마지막 문자와 일치하면 폭탄의 길이만큼 스택의 문자를 꺼내고 폭탄과 비교하여 일치하면 터뜨리고 일치하지 않으면 다시 넣는다. 이 방법을 이용하면 문자를 처음부터 끝까지 탐색하면서 CC44같은 경우가 나와도 C4를 터뜨리고 바로 C4를 터뜨릴 수 있어서 O(n)으로 문제를 해결할 수 있다. (어렵다. 이 문제를 바로 푸는 사람들은 어떻게 이런 풀이를 바로 생각해낼까 그저 대단)

 

#include <bits/stdc++.h>

using namespace std;


int main() {


    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    string s;
    string bomb;
    cin >> s >> bomb;


    stack<char> st;

    for (int i = 0; i < s.size(); i++) {
        st.push(s[i]);

        if ( s[i] == bomb[bomb.length()-1] && st.size() >= bomb.length()){
            string tempS;
            for(int j = 0 ; j < bomb.size(); j++){
                int temp = st.top();
                st.pop();
                tempS += temp;
            }
            std::reverse(tempS.begin(), tempS.end());
            if(tempS != bomb){
                for(int k = 0 ; k < bomb.size(); k++){
                    st.push(tempS[k]);
                }
            }
        }
    }

    if(st.size()==0){
        cout << "FRULA";
    }else{
        string tempS;
        int count = st.size();

        for(int j = 0 ; j < count; j++){
            int temp = st.top();
            st.pop();
            tempS += temp;
        }
        std::reverse(tempS.begin(), tempS.end());
        cout << tempS << "\n";
    }






}