백준-1541번/잃어버린 괄호 (java)

문제

수빈이는 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 문자열을 규칙에 맞게 T로 변경할 수 있는지 확인하는 문제이다.

규칙도 적어서 dfs(깊이우선탐색)를 이용해서 구현했더니 올바른 값을 얻었지만, 시간초과가 떴다.

import java.util.Scanner;

public class Main {
    static boolean check=false;
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String S = scanner.nextLine();
        String T = scanner.nextLine();
        dfs(S,T);
        if(check){
            System.out.println("1");
        }
        else{
            System.out.println("0");
        }


    }

    public static void dfs(String S, String T) {
        if(S.equals(T)){
            check = true;
            return;
        }
        if(S.length()>=T.length()){
            return;
        }
        dfs(S+'A',T);

        StringBuffer sb = new StringBuffer(S);
        sb.reverse();
        sb.append('B');
        dfs(sb.toString(),T);
    }
}

 

왜냐하면 S로 입력받을 수 있는 자릿수가 최대 999자리 이기 때문에 이걸 dfs를 이용하여 풀면 시간초과가 나는게 당연하다.

때문에 S ==> T의 방법이 아닌 T ==> S로 바꾸는 것이 현명하다.

 

T ==> S 로 바꿀때의 규칙1) T의 맨 뒷자리가 A라면 끝자리 문자 삭제2) T의 맨 뒷자리가 B라면 맨 뒷자리 지우고 문자열의 순서를 바꿔준다.

 

문제 해결

 

위에서 확인한 규칙을 이용해서 T의 문자열 길이가 S와 똑같아질때까지 T의 문자들을 삭제해준다.

while(S.length()<T.length()){
    if(T.charAt(T.length()-1)=='A'){
        T=T.substring(0,T.length()-1);
    }
    else{
        T=T.substring(0,T.length()-1);
        StringBuffer sb = new StringBuffer(T);
        sb.reverse();
        T=sb.toString();
    }
}

 

만약 삭제 결과가 S와 같으면 1을 출력하고 아니면 0을 출력하면 된다.

if(S.equals(T)){
    System.out.println("1");
}
else{
    System.out.println("0");
}

 

 

<전체 코드>

import java.util.Scanner;

public class Problem12904 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String S = scanner.nextLine();
        String T = scanner.nextLine();
        while(S.length()<T.length()){
            if(T.charAt(T.length()-1)=='A'){
                T=T.substring(0,T.length()-1);
            }
            else{
                T=T.substring(0,T.length()-1);
                StringBuffer sb = new StringBuffer(T);
                sb.reverse();
                T=sb.toString();
            }
        }

        if(S.equals(T)){
            System.out.println("1");
        }
        else{
            System.out.println("0");
        }
    }

}