백준 1309 - 동물원 JS

문제

어떤 동물원에 가로로 두칸 세로로 N칸인 아래와 같은 우리가 있다.
이 동물원에는 사자들이 살고 있는데 사자들을 우리에 가둘 때, 가로로도 세로로도 붙어 있게 배치할 수는 없다. 이 동물원 조련사는 사자들의 배치 문제 때문에 골머리를 앓고 있다.
동물원 조련사의 머리가 아프지 않도록 우리가 2*N 배열에 사자를 배치하는 경우의 수가 몇 가지인지를 알아내는 프로그램을 작성해 주도록 하자. 사자를 한 마리도 배치하지 않는 경우도 하나의 경우의 수로 친다고 가정한다.

접근법

일단 한마리도 배치하지 않는 경우도 있다는 점 + n=1일때 나올 수 있는 경우의 수가 총 3개라는 점에서 시작

일단 n=1인 칸에 올 수 있는 경우의 수는 - X X / X O / O X 총 3개만 가능함

n=2일 때

• X X 위에 붙을 수 있는 경우는 n=1일때 모든 경우의 수
• O X 위에 붙을 수 있는 경우는 X X / X O 두개만 가능
• X O 위에 붙을 수 있는 경우는 X X / O X 두개만 가능

총 7개의 가능한 경우가 있음

n=3일 때

• X X 위에 붙을 수 있는 경우는 n=2일때 3가지 경우의 수에서 가능한 모든 경우가 다 가능
• O X 위에 붙을 수 있는 경우는 n=2일때 X O 위에 붙을 수 있는 경우 + X X 위에 붙을 수 있는 경우
• X O 위에 붙을 수 있는 경우는 n=2일때 O X 위에 붙을 수 있는 경우 + X X 위에 붙을 수 있는 경우

꼭 n번째 칸 위에 올린다고 생각 안하고 첫번째 칸에서 그 밑으로 붙인다고 생각해도 상관 없을듯

코드

const fs = require("fs");
let input = parseInt(fs.readFileSync("./test.txt").toString().trim());

const zoo = Array(input + 1)
  .fill(0)
  .map((v) => Array(3));
zoo[1][0] = 1;
zoo[1][1] = 1;
zoo[1][2] = 1;
for (let i = 2; i <= input; i++) {
  zoo[i][0] = (zoo[i - 1][0] + zoo[i - 1][1] + zoo[i - 1][2]) % 9901;
  zoo[i][1] = (zoo[i - 1][0] + zoo[i - 1][2]) % 9901;
  zoo[i][2] = (zoo[i - 1][0] + zoo[i - 1][1]) % 9901;
}
const result = zoo[input].reduce((v, i) => v + i, 0);
console.log(result);