백준 12919 [자바] java A와 B 2

문제 링크: https://www.acmicpc.net/problem/12919

12919번: A와 B 2

 

▶ 문제

수빈이는 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을 출력한다.

---

▶ 해설 

DFS문제라고 생각했고, S와 T의 크기가 크지 않아서 조건없이 DFS를 진행했습니다. 하지만 시간초과가 났고 다음과 같은 방식으로 접근했습니다.

1. S의 길이를 늘리는 게 아닌 T의 길이를 줄이며, S와 같게 되는지 확인한다.

2. T의 t.charAt(length()-1) == "A"일 경우 마지막 글자를 뺀다.

3. T의 t.charAt(0)=="B" 일 경우 맨 앞글자인 "B"를 빼고 뒤집는다.

위와 같은 방식으로 진행하니 시간초과가 해결됐습니다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
 
 
public class Main {
    static int k;
    static String s, t;
    static int result;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        s = br.readLine();
        t = br.readLine();
        k = t.length();
        result = dfs(s, t);
        System.out.println(result);
 
    }
 
    public static int dfs(String s, String t) {
        if (s.length() == t.length()) {
            if (s.equals(t)) {
                return 1;
            }
            return 0;
        }
        int ans = 0;
        if (t.charAt(0) == 'B') {
            String substring = t.substring(1);
            StringBuilder sb = new StringBuilder(substring);
            String string = sb.reverse().toString();
            ans += dfs(s, string);
        }
 
        if (t.charAt(t.length() - 1) == 'A') {
            ans += dfs(s, t.substring(0, t.length() - 1));
        }
        return ans > 0 ? 1 : 0;
    }
}