|
| 1 | +/* |
| 2 | +โญ๏ธ ๋ฌธ์ ์ ๋ณด โญ๏ธ |
| 3 | +๋ฌธ์ : 72416 - ๋งค์ถ ํ๋ฝ ์ต์ํ |
| 4 | +๋ ๋ฒจ : Level 4 |
| 5 | +๋งํฌ : https://school.programmers.co.kr/learn/courses/30/lessons/72416 |
| 6 | +*/ |
| 7 | + |
| 8 | +// ANCHOR : 26.02.07 ํ์ด |
| 9 | +function solution(sales, links) { |
| 10 | + const n = sales.length; |
| 11 | + |
| 12 | + // 1) ์ธ์ ๋ฆฌ์คํธ ๊ตฌ์ฑ |
| 13 | + const children = Array.from({ length: n + 1 }, () => []); |
| 14 | + for (const [a, b] of links) { |
| 15 | + children[a].push(b); |
| 16 | + } |
| 17 | + |
| 18 | + // 2) ํ์์ํ ์ค๋น |
| 19 | + // NOTE: order๋ฅผ ์ญ์์ผ๋ก ์ฒ๋ฆฌํด์ postorder ํจ๊ณผ๋ฅผ ๋ธ๋ค. |
| 20 | + const order = []; |
| 21 | + const stack = [1]; |
| 22 | + while (stack.length) { |
| 23 | + const v = stack.pop(); |
| 24 | + order.push(v); |
| 25 | + for (const c of children[v]) { |
| 26 | + stack.push(c); |
| 27 | + } |
| 28 | + } |
| 29 | + |
| 30 | + // 3) dp ๋ฐฐ์ด ์ค๋น (1-indexed) |
| 31 | + // dp0: v๊ฐ ์ฐธ์ํ์ ๋์ ์ํฌ์ ์ฐธ์ ์ธ์์ ์ด ํ๋ฃจ ํ๊ท ๋งค์ถ์ก ์ต์๊ฐ |
| 32 | + // dp1: v๊ฐ ์ฐธ์ํ์ง ์์์ ๋์ ์ํฌ์ ์ฐธ์ ์ธ์์ ์ด ํ๋ฃจ ํ๊ท ๋งค์ถ์ก ์ต์๊ฐ |
| 33 | + const dp0 = new Array(n + 1).fill(0); |
| 34 | + const dp1 = new Array(n + 1).fill(0); |
| 35 | + |
| 36 | + // 4) Bottom-up DP ๊ณ์ฐ (order๋ฅผ ์ญ์์ผ๋ก ๋๋ฉด์ ํ์์ํ) |
| 37 | + for (let i = order.length - 1; i >= 0; i--) { |
| 38 | + const v = order[i]; |
| 39 | + const childs = children[v]; |
| 40 | + |
| 41 | + // case 1) ๋ฆฌํ์ธ ๊ฒฝ์ฐ (์์ ํ์์ธ ๊ฒฝ์ฐ) |
| 42 | + // - dp0[v] = sales[v] (๋ณธ์ธ์ด ์ฐธ์) |
| 43 | + // - dp1[v] = 0 (๋ณธ์ธ ๋ถ์ฐธ. ์๋ ํ์ด ์๊ธฐ ๋๋ฌธ์ ์์ ์ค ํ๋ช
์ด ๊ผญ ์ฐธ์ํด์ผ ํ๋ค๋ ๊ฐ์ ๊ฐ ์๋ค. ๋ถ์ฐธํ๋ฉด ๋.) |
| 44 | + if (childs.length === 0) { |
| 45 | + dp1[v] = sales[v - 1]; |
| 46 | + dp0[v] = 0; |
| 47 | + continue; |
| 48 | + } |
| 49 | + |
| 50 | + // case 2) ํ์ฅ์ด๊ณ ํ์์ธ ๊ฒฝ์ฐ (ํ์ ํ์ด ์์) |
| 51 | + // a) ์ฐธ์ํ๋ ๊ฒฝ์ฐ : ๊ฐ์ ํ์ ๋ค๋ฅธ ํ์๋ค์ ์ฐธ์ํ๋ ์ํ๋ ์๊ด ์์ |
| 52 | + // -> dp1[v] = sales[v] + โ min(dp0, dp1) |
| 53 | + // b) ๋ถ์ฐธํ๋ ๊ฒฝ์ฐ : ๊ฐ์ ํ์ ๋ค๋ฅธ ํ์ ์ค ์ ์ด๋ ํ๋ช
์ ์ฐธ์ํด์ผ ํจ |
| 54 | + // -> ์ฐ์ ๋ชจ๋ ํ์๋ค์์ ๊ฐ์ฅ ์ผ ์ํ๋ฅผ ๋ํ ๋ค (sumMin) |
| 55 | + // -> ์ฐธ์์ผ๋ก ๋๋ ธ์ ๋ ๋น์ฉ์ด ๊ฐ์ฅ ์ ๊ฒ ์ถ๊ฐ๋๋ ์ธ์์ ์ฐธ์์ผ๋ก ๋๋ฆฐ๋ค. (extraMin) |
| 56 | + // -> dp0[v] = sumMin + extraMin |
| 57 | + let sumMin = 0; |
| 58 | + let extraMin = Infinity; |
| 59 | + |
| 60 | + for (const c of childs) { |
| 61 | + const m = Math.min(dp0[c], dp1[c]); // ์ฐธ์ํ๋ ๊ฒ๊ณผ ์ํ๋ ๊ฒ ์ค ์ต์๊ฐ |
| 62 | + sumMin += m; |
| 63 | + |
| 64 | + const extra = dp1[c] - m; // (์ฐธ์์ผ๋ก ๋๋ ธ์ ๋ ์ถ๊ฐ๋๋ ๋น์ฉ) |
| 65 | + if (extra < extraMin) { |
| 66 | + extraMin = extra; |
| 67 | + } |
| 68 | + } |
| 69 | + |
| 70 | + dp1[v] = sales[v - 1] + sumMin; |
| 71 | + dp0[v] = sumMin + extraMin; |
| 72 | + } |
| 73 | + |
| 74 | + return Math.min(dp0[1], dp1[1]); |
| 75 | +} |
0 commit comments