본문 바로가기
dev/코딩테스트

[BOJ 9935] 문자열 폭발

by dev-everyday 2024. 12. 16.
반응형

문제

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

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

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

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

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

 

입력

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

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

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

 

출력

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

 

풀이

stack을 사용할 필요 없이 string만 이용해서 해당 문제를 풀 수 있다.

기존 string과 비교 string인 cmp_str을 입력받는다.

먼저 for문을 돌면서 str의 값을 넣어주고 만약 ans의 문자열 길이가 비교 문자열 길이보다 길다면 비교 문자열 길이만큼 substr을 통해서 자른 후 cmp_str과 비교해준다.

만약 둘이 같다면 ans에서 해당 길이만큼 문자열을 삭제하면 된다.

 

ans.substr(ans.size() - cmp_len) 은 ans에서 ans.size() - cmp_len 인덱스에서 뒷 부분 전부를 추출한다는 뜻으로 뒤에 인자를 생략하면 문자열 끝까지 포함된다.

 

코드

#include <iostream>
#include <string>
using namespace std;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	string str, cmp_str;
	cin >> str >> cmp_str;
	int cmp_len = cmp_str.size();

	string ans = "";
	for (char c : str) {
		ans.push_back(c);

		if (ans.size() >= cmp_len && ans.substr(ans.size() - cmp_len) == cmp_str) {
			ans.erase(ans.size() - cmp_len);
		}
	}

	if (ans.empty()) {
		cout << "FRULA";
	}
	else {
		cout << ans;
	}

	return 0;
}

 

반응형

'dev > 코딩테스트' 카테고리의 다른 글

[BOJ - 2178] 미로 탐색  (2) 2024.12.16
[BOJ - 10845] 큐  (3) 2024.12.16
[BOJ - 2800] 괄호 제거  (4) 2024.12.16
[BOJ - 9663] N-Queen  (0) 2024.12.13
[BOJ - 2504] 괄호의 값  (0) 2024.12.13