[백준 알고리즘] 12904번 자바(Java) A와 B

문제

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

 

풀이

반대로 T에서 맨뒤에 글자를 하나씩 제거하면 S가 나오는지를 확인해서 답을 알 수있다.
끝이 A면 A제거만

B면 B제거하고 문자열 반대로 해야한다. 

덱을 이용해서 데이터를 다 넣어논 후에 앞과 뒤 값 제거를 통해 T를 줄여나가고

그것을 S와 비교한다. 시간복잡도 상으로 덱의 처음와 마지막index에 접근 및 제거는 O(1)이므로

T의 길이만큼만의 시간복잡도가 발생한다.

 

예제 입력 예시

B
ABBA

예제 출력

1

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.Deque;

//12904번
//S에 A와B를 추가하면서 T에 맞출생각하지말고 T에서 마지막글자를 통해 T를 S에 맞출생각을 한다

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();

        Deque<Character> deque = new ArrayDeque<>();
        for (Character ch : T.toCharArray()) {
            deque.addLast(ch);
        }

        //front true면 정방향(마지막 글자확인), false면 반대방향(첫번째 글자확인)
        boolean front = true;
        //덱 사이즈가 S랑 같아질때까지 T의 마지막글자 확인
        while(deque.size()>S.length()){
            if(front){  //정방향
                if(deque.peekLast() == 'A'){
                    deque.removeLast();
                }
                else{
                    deque.removeLast();
                    front = false;
                }
            }
            else{  //반대방향
                if(deque.peekFirst() == 'A'){
                    deque.removeFirst();
                }
                else{
                    deque.removeFirst();
                    front = true;
                }
            }
        }

        //StringBuffer사용해서 덱에서 흩어져있는 문자들 합쳐서 문자열 맞추기
        StringBuffer sb = new StringBuffer();
        
        if (front) {  //정방향이면 앞부터 순서대로
            while (!deque.isEmpty()) {
                sb.append(deque.removeFirst());
            }
        }
        else {  //반대방향이면 뒤부터 순서대로
            while (!deque.isEmpty()) {
                sb.append(deque.removeLast());
            }
        }

        //해석된 T와 S 비교
        if (sb.toString().equals(S)) System.out.println("1");
        else System.out.println("0");
    }
}