[baekjoon] #1149 RGB거리 (Node.js)

📖 문제

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

☃️ 풀이

한참을 고민하던 문제. 질문 게시판에 올려진 글들을 보고 힌트를 얻었다.거의 답을 확인한 수준이지만..

각 단계마다 RGB 색상으로 집을 색칠하는 비용이 각각 주어지고, 각 비용 각각에 대해서 인덱스 번호가 다른 다음 단계 배열의 항목들 중 가장 작은 값을 더한다.

처음에는 위 과정을 switch문을 통해 일일히 작성해야 하나.. 했다.
근데 { 0: [1, 2], 1: [0, 2], 2: [0, 1] } 이렇게 객체로 선택할 배열의 인덱스를 미리 정해주는 방법이 있었다. 생각해보면 참 간단한데 왜 난 이런 걸 생각해내지 못하는지.. 또 하나 배워간다.

아무튼 위 과정을 반복하며 첫번째 RGB값 각각에서 출발했을 경우 얻을 수 있는 최솟값을 구하고, 다음 단계로 넘어갈 때마다 dp 값을 갱신해준다. 모든 과정이 끝난 뒤, dp 배열의 최솟값을 반환한다.

💽 소스코드

const input = require('fs').readFileSync('input.txt').toString().split('\n');
const n = Number(input.shift());
const nx = { 0: [1, 2], 1: [0, 2], 2: [0, 1] };
let dp = [0, 0, 0];

for (let i = 0; i < n; i++) {
  const arr = input.shift().split(' ').map(Number);

  for (let j = 0; j < 3; j++) {
    arr[j] += Math.min(dp[nx[j][0]], dp[nx[j][1]]);
  }

  dp = arr;
}

console.log(Math.min(...hold));