(22.07.27) (Baekjoon) 12904.A와 B

12904번: A와 B

시간 제한 : 2초

메모리 제한 : 512MB

 

문제

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

예제 입력 1 복사

B
ABBA

예제 출력 1 복사

1

예제 입력 2 복사

AB
ABB

예제 출력 2 복사

0

 

풀이

a가 시작 문자

b가 완성 문자라고 가정했을 때,

b에서 문자를 줄여가면서 a가 되는지 비교하면 쉽게 해결할 수 있다.

 

문제에선 a에 문자를 추가하는 방법이

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

라고 명시되어 있는데, 이를 바꿔 b에서 a로 될 수 있는 방법을 생각해보자면,

• 문자열의 뒤에 A가 있다면, A를 제거한다.
• 문자열의 뒤에 B가 있다면, B를 제거하고 문자열을 뒤집는다.

다음과 같은 내용을 유념하면서 코드를 짜보면 문제없이 해결할 수 있다.

Java에선 StringBuilder(혹은 StringBuffer)를 활용할 수 있으니 이를 유념하자.

StringBuilder.append(String str) : StringBuilder에 특정 문자를 추가해주는(이미 있을 경우 끝에 붙여주는) 메서드

StringBuilder.reverse() : 문자열을 뒤집어주는 메서드

StringBuilder.delete(int start, int end) : StringBuilder에 있는 문자에서 (start ~ end-1)까지의 문자를 제거하는 메서드

위 세 메서드를 잘 활용하면 어렵지 않게 해결할 수 있는 문제였다

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {
	
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String a = br.readLine();
		StringBuilder sb = new StringBuilder();
		sb.append(br.readLine());
		while(sb.toString().length() > a.length()) {
			if(sb.charAt(sb.length()-1) == 'A') {
				sb.delete(sb.length()-1, sb.length());
			}
			else if(sb.charAt(sb.length()-1) == 'B') {
				sb.delete(sb.length()-1, sb.length());
				sb.reverse();

			}
		}
		if(sb.toString().equals(a)) {
			System.out.println(1);
		}
		else {
			System.out.println(0);
		}
	}

}