코딩테스트/그리디(Greedy)

[Java] 백준 12904번 : A와 B

sujin7837 2022. 6. 14. 00:44
반응형

문제

수빈이는 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

 

소스코드

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

public class Main {

	private static BufferedReader br;
	
	private static String S, T;
	
	public static void main(String[] args) throws IOException {
		br=new BufferedReader(new InputStreamReader(System.in));
		S=br.readLine();
		T=br.readLine();

		for(int i=T.length()-1;i>=S.length();i--) {	// T 문자의 맨 끝부터 거꾸로 탐색
			if(T.charAt(i)=='B') {	// B인 경우 B를 자르고 문자열을 뒤집음
				T=T.substring(0, i);
				StringBuilder sb=new StringBuilder(T);
				T=sb.reverse().toString();
				
			} else T=T.substring(0, i);	// A인 경우 A를 자르고 문자열을 뒤집지 않음
		}
		
		if(!S.equals(T)) System.out.println(0);	// 최종 두 문자열이 같으면 1, 다르면 0
		else System.out.println(1);
	}

}

T -> S 로 거꾸로 문자열을 바꾸며 마지막에 두 문자열이 같은지 확인하면 되는 문제였습니다.

 

T의 가장 끝 문자가 'A'인 경우, 문자열을 뒤집지 않음

T의 가장 끝 문자가 'B'인 경우, 문자열을 뒤집음

반응형