[백준 / BAEKJOON] 9935번 문자열 폭발 - C++

문제 원문

조건

시간 제한 : 2초 (추가 시간 없음)
메모리 제한 : 128 MB

문제

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

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

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

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

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

입력

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

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

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

출력

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

예제

풀이

처음엔 KMP 알고리즘을 사용해서 풀었으나 시간초과가 되었습니다.

이 문제 같은 경우엔 폭발문자열(검증 문자열에 포함되어 있으면 제거되는 문자열)의 길이가 최대 36으로 KMP 알고리즘을 사용하는 것보다 맨 앞이나 맨 뒤의 문자가 같을 때 폭발 문자열이 일치하는지 검증하는 것이 더 빠르게 처리됩니다.

저는 결과 문자열을 쌓아가면서 처리를 했고 폭발문자열이 제거된 이후 문자열을 검증할 때도 뒷 문자가 같을 때 검증하는 것이 시퀀스상 쉽고 빠르게 동작하기 때문에 맨 뒤의 문자가 같을 때 검증하도록 처리하였습니다.

풀이 방법은 다음과 같습니다.

1. 결과문자열 변수를 검사문자열을 생성자로 선언하고 해당 문자열 인덱스를 0으로 선언
2. for문으로 검사문자열을 순회한다
3. 순회하면서 결과문자열의 인덱스에 검사문자열의 문자를 넣는다
4. 만약 넣은 문자와 폭발문자열의 마지막 문자가 같고 결과문자열의 길이가 폭발문자열의 길이보다 크거나 같다면 문자열이 일치하는지 검증한다.
5. 정확히 일치할 경우 폭발문자열의 길이만큼 결과문자열의 인덱스를 감소시킨다.

성공 코드

#include <iostream>
#include <string>
#include <vector>

using namespace std;

//define

using ull = unsigned long long ;
using location =  pair<int, int> ;
using matrix = vector<vector<int>>;

//custum function
string solution(string mainStr, string subStr)
{
	int mainSize = mainStr.size();
	int subSize = subStr.size();

	//main을 복사한 result를 폭탄이 터졌을때 인덱스를 이동시키는 것으로 속도를 향상시킨다.
	string resultStr(mainStr);
	int mainIndex(0), resultIndex(0);

	for(mainIndex; mainIndex < mainSize; ++mainIndex)
	{
		resultStr[resultIndex] = mainStr[mainIndex];
		if (resultIndex >= subSize - 1 && mainStr[mainIndex] == subStr[subSize - 1])
		{
			bool same = true;
			for (int subIndex(0); subIndex < subSize; ++subIndex)
			{
				if (subStr[subIndex] != resultStr[resultIndex - (subSize - subIndex) + 1])
				{
					same = false;
					break;
				}
			}
			if (same)
			{
				resultIndex -= subSize;
			}
		}
		++resultIndex;
	}

	return resultStr.substr(0, resultIndex);
}

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	////////////////////////////////////
	//Declaration
	string inputStr(""), bombStr("");
	cin >> inputStr;
	cin >> bombStr;

	//Solution
	string resultStr = solution(inputStr, bombStr);


	//Output
	if (resultStr.size() != 0)
	{
		cout << resultStr << '\n';
	}
	else
	{
		cout << "FRULA\n";
	}

	return 0;
}

결과

Github - github.com/ledpear