[BOJ] RGB 거리 | DP

Urther·2021년 10월 23일
0

알고리즘

목록 보기
20/41
post-thumbnail

Problem | RGB 거리


RGB거리에는 집이 N개 있다. 거리는 선분으로 나타낼 수 있고, 1번 집부터 N번 집이 순서대로 있다.

집은 빨강, 초록, 파랑 중 하나의 색으로 칠해야 한다. 각각의 집을 빨강, 초록, 파랑으로 칠하는 비용이 주어졌을 때, 아래 규칙을 만족하면서 모든 집을 칠하는 비용의 최솟값을 구해보자.

1번 집의 색은 2번 집의 색과 같지 않아야 한다.
N번 집의 색은 N-1번 집의 색과 같지 않아야 한다.
i(2 ≤ i ≤ N-1)번 집의 색은 i-1번, i+1번 집의 색과 같지 않아야 한다.

✨ 접근 방식

입력 예제
3
26 40 83
49 60 57
13 89 99

만약, 2번째 집(i=1) 을 Green 으로 칠하는 경우는 1번째 집 (i=i-1)에서 Red 혹은 Blue로 칠하는 값의 최솟값 + (i=1)이 Green으로 칠하는 비용을 더해주면 된다.

- ✔️ 전체코드

  • 풀어놓고 본건데 이렇게 for문을 많이 쓸 필요는 없었을 것 같다는 생각이 들었다. (어차피 RGB 3개의 색상이니까 !! )
let fs = require("fs");
//let input = fs.readFileSync("/dev/stdin").toString().split("\n");
let input = fs.readFileSync("./input.txt").toString().split("\n");
// <--- 입력 예제 --->
// 3
// 26 40 83
// 49 60 57
// 13 89 99
// <----------------->

N=Number(input[0]);
color=new Array(N);
for(let i=0;i<N;i++) color[i]=input[i+1].split(' ');
// color라는 box에 얼마인지 넣기

let dp=Array.from(Array(N),()=>Array(3).fill(0));
for(let i=0;i<3;i++) dp[0][i]=Number(color[0][i]);


// dp실행
for(let i=1;i<N;i++){
  for(let j=0;j<3;j++){
    if(j===0) dp[i][j]=Math.min(dp[i-1][j+1],dp[i-1][j+2])+parseInt(color[i][j]);
    else if(j===1) dp[i][j]=Math.min(dp[i-1][j-1],dp[i-1][j+1])+parseInt(color[i][j]);
    else dp[i][j]=Math.min(dp[i-1][j-1],dp[i-1][j-2])+parseInt(color[i][j]);
  }
}

// 마지막 집 색칠 중 가장 작은 수를 골라야한다.
let answer=Number.MAX_SAFE_INTEGER;
for(let i=0;i<3;i++) answer=Math.min(answer,dp[N-1][i]);

console.log(answer);

DP는 정말루 어렵다.. 그래두 화이이이팅 ~ ~ ~ ! !

profile
이전해요 ☘️ https://mei-zy.tistory.com

0개의 댓글