solved(python): baekjoon 17404

Author: himitery

baekjoon/python/17404/__init__.py

@@ -0,0 +1,29 @@
+import math
+import sys
+
+read = lambda: sys.stdin.readline().rstrip()
+
+
+class Problem:
+    def __init__(self):
+        self.n = int(read())
+        self.data = [list(map(int, read().split())) for _ in range(self.n)]
+
+    def solve(self) -> None:
+        print(min(*[self.find_min_cost(color) for color in range(3)]))
+
+    def find_min_cost(self, start_color: int) -> int:
+        dp = [[math.inf for _ in range(3)] for _ in range(self.n)]
+        dp[0][start_color] = self.data[0][start_color]
+
+        for row in range(1, self.n):
+            dp[row][0] = min(dp[row - 1][1], dp[row - 1][2]) + self.data[row][0]
+            dp[row][1] = min(dp[row - 1][0], dp[row - 1][2]) + self.data[row][1]
+            dp[row][2] = min(dp[row - 1][0], dp[row - 1][1]) + self.data[row][2]
+
+        dp[-1][start_color] = math.inf
+        return int(min(dp[-1]))
+
+
+if __name__ == "__main__":
+    Problem().solve()

baekjoon/python/17404/main.py

@@ -0,0 +1,90 @@
+[
+  {
+    "input": [
+      "3",
+      "26 40 83",
+      "49 60 57",
+      "13 89 99"
+    ],
+    "expected": [
+      "110"
+    ]
+  },
+  {
+    "input": [
+      "3",
+      "1 100 100",
+      "100 1 100",
+      "100 100 1"
+    ],
+    "expected": [
+      "3"
+    ]
+  },
+  {
+    "input": [
+      "3",
+      "1 100 100",
+      "100 100 100",
+      "1 100 100"
+    ],
+    "expected": [
+      "201"
+    ]
+  },
+  {
+    "input": [
+      "6",
+      "30 19 5",
+      "64 77 64",
+      "15 19 97",
+      "4 71 57",
+      "90 86 84",
+      "93 32 91"
+    ],
+    "expected": [
+      "208"
+    ]
+  },
+  {
+    "input": [
+      "8",
+      "71 39 44",
+      "32 83 55",
+      "51 37 63",
+      "89 29 100",
+      "83 58 11",
+      "65 13 15",
+      "47 25 29",
+      "60 66 19"
+    ],
+    "expected": [
+      "253"
+    ]
+  },
+  {
+    "input": [
+      "4",
+      "3 2 1",
+      "1 2 2",
+      "2 2 1",
+      "2 1 2"
+    ],
+    "expected": [
+      "4"
+    ]
+  },
+  {
+    "input": [
+      "5",
+      "9 2 8",
+      "8 6 5",
+      "4 8 5",
+      "3 8 2",
+      "6 8 8"
+    ],
+    "expected": [
+      "19"
+    ]
+  }
+]

baekjoon/python/17404/sample.json

@@ -0,0 +1,36 @@
+import json
+import os.path
+import unittest
+from io import StringIO
+from unittest.mock import patch
+
+from parameterized import parameterized
+
+from main import Problem
+
+
+def load_sample(filename: str):
+    path = os.path.join(os.path.dirname(os.path.abspath(__file__)), filename)
+
+    with open(path, "r") as file:
+        return [(case["input"], case["expected"]) for case in json.load(file)]
+
+
+class TestCase(unittest.TestCase):
+    @parameterized.expand(load_sample("sample.json"))
+    def test_case(self, case: str, expected: list[str]):
+        # When
+        with (
+            patch("sys.stdin.readline", side_effect=case),
+            patch("sys.stdout", new_callable=StringIO) as output,
+        ):
+            Problem().solve()
+
+            result = output.getvalue().rstrip()
+
+        # Then
+        self.assertEqual("\n".join(expected), result)
+
+
+if __name__ == "__main__":
+    unittest.main()