🎲 백준 2011번 암호코드

백준 2011번

🧸 암호를 해석할 수 없는 경우를 처리하는 것에서 헤맸다.

코드

const fs = require('fs'); 
const input = fs.readFileSync('/dev/stdin').toString().trim().split('');

const dp = new Array(input.length + 1);
dp[0] = 1;
//0으로 시작할 경우
if (Number(input[0]) <= 0) {
  return console.log(0);
}
dp[1] = 1;

for (let i = 2; i <= input.length; i++) {
  const two = Number(input[i - 2] + input[i - 1]);
  const one = Number(input[i - 1]);
  //암호가 잘못된 경우 미리 처리
  if (one === 0 && (two > 26 || input[i - 2] === "0")) {
    return console.log(0);
  }
  
  if (two <= 26 && two >= 10) {
    if (one === 0) {
      dp[i] = dp[i - 2];
    } else {
      dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000;
    }
  } else {
    dp[i] = dp[i - 1];
  }
}

console.log(dp[input.length] % 1000000);