Algorithm

[BOJ] A와 B 2 12919번 C++

따봉치치 2024. 5. 30. 14:39

 

 

 

문제

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

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

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

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

입력

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

출력

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

 

 

접근 방식

 

S로 T를 만들 수 있는 모든 경우의 수를 구하려고 하니 시간초과가 발생했다

따라서 T로 S를 만들 수 있는지 역으로 구했다

1. 문자열의 가장 마지막 문자가 A 라면 해당 문자열 제외하기

2. 문자열의 가장 첫번째 문자가 B 라면 문자열을 뒤집고 해당 문자열 제외하기

 

이렇게 코드를 작성했더니 시간초과가 발생하지 않고 잘 동작하게 되었다!

 

 

코드

 

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

string S,T;
bool isMake = false;

void sol(string str) {
    if(str.length() == S.length()) {
        if(str == S) isMake = true;
        return;
    }
    if(isMake) return;

    if(str[str.length()-1] == 'A') sol(str.substr(0,str.length()-1));
    if(str[0] == 'B') {
        reverse(str.begin(), str.end());
        sol(str.substr(0,str.length()-1));
    }
    
    

}

int main() {
    ios::sync_with_stdio(false); cin.tie(0);
    cin>>S>>T;

    sol(T);
    if(isMake) cout<<1;
    else cout<<0;
}