BOJ/Node.js

[BOJ / Node.js] 12919. A와 B 2

도리닥닥 2023. 2. 27. 01:50
728x90

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

 

12919번: A와 B 2

수빈이는 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의 길이 ≤ 49, 2 ≤ T의 길이 ≤ 50, S의 길이 < T의 길이)

출력

S를 T로 바꿀 수 있으면 1을 없으면 0을 출력한다.

예제 입력 1

A
BABA

예제 출력 1

1

 

Solve

  • 모든 경우를 체크하기 위해서 재귀를 사용하였고 base case를 s와 t의 길이로 설정하였습니다.
  • 주어진 경우를 통과할 경우 answer 배열에 담고 answer에 t가 포함이 되어있다면 1, 아니라면 0을 출력하도록 하였습니다.

Problem

  • 재귀함수 속 문자열 s가 target에 포함되지 않는 경우와 뒤집힌 s가 target에 포함되지 않는 경우를 return해 주지 않을경우, 불필요한 연산이 되어 시간초과가 발생했었습니다. 

Code

const [s,t] = require("fs").readFileSync("/dev/stdin").toString().trim().split("\n");

// 일단 길이가 맞아야함 => base case
const answer = [];
const solution = (s, target) => {
    if(!target.includes(s) && !target.includes([...s].reverse().join(''))) return;
    if(s.length === target.length) {
        return answer.push(s);
    }
    
    solution(s + "A", target);
    solution([...s+"B"].reverse().join(''), target)
}

solution(s, t);

console.log(answer.includes(t) ? 1 : 0);