Dot Algo∙ DS/PS
[BOJ] 백준 12919번 A와 B 2 (Java)
루지
2022. 3. 8. 12:58
#12919 A와 B 2
난이도 : 골드 5
유형 : 문자열 / 재귀
▸ 문제
수빈이는 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을 출력한다.
문제 풀이
모든 경우의 수를 조사하면 최대 2^50의 경우를 조사해줘야 한다. 그래서 나누어지는 가지의 수는 2가지만 고려하여 분기해줘야 한다. 맨 뒤 문자열이 A인 경우와 앞의 문자열이 B인 경우이다.
dfs 재귀 방식으로 탐색하여 풀어줬다. dfs(s, t)
- t 맨 뒤의 문자열 A인 경우, A를 제거하고 재귀 탐색한다.
- t 맨 앞의 문자열이 B인 경우, B를 제거하고 문자열을 뒤집은 후 재귀 탐색한다.
- s == t의 길이가 같아지는 경우, 두 문자열이 일치하는지 조사한다.
- 한 가지의 경우라도 일치하는 경우 true를 반환하고 하나도 일치하지 않은 경우 false를 반환한다.
문자열 뒤집는 것은 StrigBuilder의 reverse()를 이용하면 간단하게 뒤집을 수 있다.
풀이 코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main {
public static void main(String[] args) throws IOException{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String s = br.readLine();
String t = br.readLine();
System.out.println(dfs(s,t)? 1: 0);
}
static boolean dfs(String s, String t) {
boolean flag = false;
if(s.length() >= t.length()) {
if(s.equals(t)) {
return flag = true;
}
return flag = false;
}
if(t.charAt(t.length()-1) == 'A') {
flag |= dfs(s, t.substring(0, t.length()-1));
}
if(t.charAt(0) == 'B'){
flag |= dfs(s, new StringBuilder().append(t).reverse().substring(0, t.length()-1).toString());
}
return flag;
}
}