[백준] 12094 - A와 B

https://www.acmicpc.net/problem/12904

12904번: 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을 출력한다.

 

---

풀이

 

쉬운 그리디(Greedy) 알고리즘 문제이다. 처음 접근할 때 아무 생각없이 백트래킹으로 전체 경우의 수를 구하다보니 시간 초과가 발생했다. memoization을 시도해봤지만 메모리 초과로 다른 풀이 방법을 생각해보았다.

 

결론적으로 S에서 T를 만드는 방법이 아닌, T에서 S를 만들 수 있는지 체크를 해주었다. 문제에 제시된 방법을 통해서 역으로 구해가면 검증할 수 있다.

 

로직

• T의 마지막 글자가 A인 경우
  • A를 버림 ➡️ String.substring() 사용
• T의 마지막 글자가 B인 경우
  • T를 뒤집고, 첫 인덱스로 간 B를 버림 ➡️ StringBuilder와 String.substring() 사용
• 만약, T가 모두 버려지고 길이가 0이 된다면, T로 S를 만들 수 없다.

 

처음부터 그리디로 접근했으면 무난하게 풀 수 있었던 문제인데, 항상 완전 탐색을 우선해서 생각하다 보니까 몇 번 틀리고 시작하게 된다. 조금 더 연습이 필요할 듯 하다.

 

 

 

전체 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;

public class Main {

    public static Set<String> cache;
    public static String s, t;

    public static void input() throws IOException{

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        s = br.readLine();
        t = br.readLine();

        cache = new HashSet<>();
    }

    public static boolean isMade() {

        StringBuilder sb;

        while(!s.equals(t)) {
            if(t.length() == 0) return false;
            if(t.charAt(t.length() - 1) == 'A') {
                t = t.substring(0, t.length() - 1);
            }else if(t.charAt(t.length() - 1) == 'B') {
                sb = new StringBuilder(t);
                t = sb.reverse().substring(1);
            }else return false;
        }
        return true;
    }

    public static void makeSameWord() {

        System.out.println(isMade() ? 1 : 0);
    }

    public static void main(String[] args) throws IOException {

        input();
        makeSameWord();
    }
}