[백준 c++] 9935 문자열 폭발

jw·2022년 10월 31일
0

백준

목록 보기
62/141
post-thumbnail

문제 설명

https://www.acmicpc.net/problem/9935
상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.
폭발은 다음과 같은 과정으로 진행된다.
문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.
상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.
폭발 문자열은 같은 문자를 두 개 이상 포함하지 않는다.

입력
첫째 줄에 문자열이 주어진다. 문자열의 길이는 1보다 크거나 같고, 1,000,000보다 작거나 같다.
둘째 줄에 폭발 문자열이 주어진다. 길이는 1보다 크거나 같고, 36보다 작거나 같다.
두 문자열은 모두 알파벳 소문자와 대문자, 숫자 0, 1, ..., 9로만 이루어져 있다.

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

아이디어

처음엔 문자열 for문으로 풀었는데 시간초과가 났다. 이유는 매번
폭발이 발생하면 해당 문자열을 자르고, 그 전까지 문자열은 붙이고 다시
문자열 앞에서 부터 탐색하는 과정을 반복했기 때문이다.
그래서 스택으로도 풀어봤는데 정답처리됐지만 이런 저런 예외처리를 하느라 코드가 복잡했다.

문자열 erase 기반 pivot 알고리즘을 이용해서 풀면 간단하게 풀 수 있었다.
ret에 하나씩 더해가면서 ret 길이가 폭발 문자열보다 크거나 같아지면 substr로 문자열 비교를 해서 erase한다.

전체 코드

#include <iostream>
#include <algorithm>
using namespace std;
string s, t, ret;
int main()
{
    cin >> s >> t;
    for (char a : s)
    {
        ret += a;
        if (ret.size() >= t.size() && ret.substr(ret.size() - t.size(), t.size()) == t)
        {
            ret.erase(ret.end() - t.size(), ret.end());
        }
    }
    if (!ret.size())
        cout << "FRULA"
             << "\n";
    else
        cout << ret << "\n";
}
profile
다시태어나고싶어요

0개의 댓글