본문 바로가기

Dot Algo∙ DS/PS

[BOJ] 백준 12904번 A와 B (Java)

    #12904 A와 B

    난이도 : 골드 4

    유형 : 그리디 / 문자열

     

    12904번: A와 B

    수빈이는 A와 B로만 이루어진 영어 단어가 존재한다는 사실에 놀랐다. 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다. 이런 사실에 놀란 수

    www.acmicpc.net

    ▸ 문제

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

     

    문제 풀이  

    브루트포스 알고리즘으로 재귀 탐색으로 S문자열을 매 depth마다 'str+"A"', 'str.reverse()+"B"'로  최대 길이가 1000인 문자열을 매번 비교해가면서 연산하면 시간초과가 발생한다. DP를 사용하기에는 문자열이라서 데이터를 다루기 까다로워진다. 그래서 그리디 알고리즘을 사용해 최적해 탐색을 해줘야 한다.

     

    매 문자열을 추가하며 S > T를 탐색하는 방법 대신에 T > S로 문자열을 줄여가며 탐색하는 방법을 사용해야 한다. 코드로만 봐도 전자의 방법은 문자열 길이 1이 늘어날 때 마다 가지수가 2개씩 늘어나고 후자인 경우 끝자리의 문자만 제거하면 되므로 더 적은 경우의 수로 최적해 탐색을 할 수 있다.

     

    (왼)S>T, (오른) T>S

     

    설계

    1. (t의 길이 > s의 길이)일 때까지 while반복문을 돌린다. while(tb.length()>sb.length())
      1. t의 끝자리가 'A'이면, t의 끝자리 'A'를 삭제한다.  
      2. t의 끝자리가 'B'이면, t의 끝자리 'B'를 삭제하고 문자열을 뒤집는다.
    2. (t의 길이 == s의 길이)이면, while반복문을 종료하고 두 문자열이 일치하면 1, 아니면 0 을 출력한다. if(tb.toString().equals(sb.toString())) flag =1;

     

    풀이 코드 

    import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStreamReader;
    
    public class Main {
    
    	static int res;
    	public static void main(String[] args) throws IOException{
    		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    		
    		StringBuilder sb = new StringBuilder(br.readLine());
    		StringBuilder tb = new StringBuilder(br.readLine());
    		
    		int flag=0;
    		while(tb.length()>sb.length()) {
    			if(tb.charAt(tb.length()-1) == 'A') {
    				tb.delete(tb.length()-1, tb.length());
    			}else {
    				tb.delete(tb.length()-1, tb.length());
    				tb.reverse();
    			}
    			
    			
    			if(tb.length() == sb.length()) {
    				if(tb.toString().equals(sb.toString())) flag =1;
    				break;
    			}
    		}
    		System.out.println(flag);
    	}
    }