C++ 백준 12904번: A와B

https://www.acmicpc.net/problem/12904

12904번: A와 B

문제

수빈이는 A와 B로만 이루어진 영어 단어가 존재한다는 사실에 놀랐다. 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다.

이런 사실에 놀란 수빈이는 간단한 게임을 만들기로 했다. 두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다. 문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.

• 문자열의 뒤에 A를 추가한다.
• 문자열을 뒤집고 뒤에 B를 추가한다.

주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오. 

입력

첫째 줄에 S가 둘째 줄에 T가 주어진다. (1 ≤ S의 길이 ≤ 999, 2 ≤ T의 길이 ≤ 1000, S의 길이 < T의 길이)

출력

S를 T로 바꿀 수 있으면 1을 없으면 0을 출력한다.

---

풀이

아이디어 생각하는데 시간이 조금 걸렸는데 구현은 아주 쉽게 뚝딱

S를 T로 만드는 것을 반대로 T를 S로 만드는 것으로 바꿔서 생각했다. 그에 따라 조건도 문자열 맨 뒤가 A이면 삭제 B이면 삭제하고 뒤집기를 해주었다. 그리고 두 문자열의 길이가 같아지면 비교해서 1 또는 0 출력!

 

아마 이 생각 못했으면  정말 오래걸렸을 듯,,, 

---

#include <bits/stdc++.h>
using namespace std;

int main() {
	string tmp, ans;
	cin >> tmp;
	cin >> ans;

	while (true) {
		if (tmp.size() == ans.size()) {
			if (tmp == ans) {
				cout << 1 << '\n';
				return 0;
			}
			else {
				cout << 0 << '\n';
				return 0;
			}
		}

		int num = ans.size();
		if (ans[num - 1] == 'A') { //그냥 삭제
			ans.pop_back();
		}
		else { //  삭제하고 reverse
			ans.pop_back();
			reverse(ans.begin(), ans.end()); 
		}
	}
}