백준 1932번 정수 삼각형

문제 링크: https://www.acmicpc.net/problem/1932

Author: skysign

백준 1932번 정수 삼각형/main.py

@@ -0,0 +1,25 @@
+import sys
+from typing import List
+
+
+def solve():
+    n = int(sys.stdin.readline().strip())
+    dts: List[List[int]] = [[0 for _ in range(n + 1)] for _ in range(n + 1)]
+    dp: List[List[int]] = [[0 for _ in range(n + 1)] for _ in range(n + 1)]
+
+    for row_idx in range(1, n + 1):
+        line = list(map(int, sys.stdin.readline().strip().split(' ')))
+        for col_idx in range(1, row_idx + 1):
+            dts[row_idx][col_idx] = line[col_idx - 1]
+
+    dp[1][1] = dts[1][1]
+
+    for row_idx in range(2, n + 1):
+        for col_idx in range(1, row_idx + 1):
+            dp[row_idx][col_idx] = dts[row_idx][col_idx] + max(dp[row_idx - 1][col_idx], dp[row_idx - 1][col_idx - 1])
+
+    print(max(dp[n]))
+
+
+if __name__ == '__main__':
+    solve()

백준 1932번 정수 삼각형/test1.txt

@@ -0,0 +1,6 @@
+5
+7
+3 8
+8 1 0
+2 7 4 4
+4 5 2 6 5

백준 1932번 정수 삼각형/test1_answer.txt

@@ -0,0 +1 @@
+30

백준 1932번 정수 삼각형/test_main.py

@@ -0,0 +1,23 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+
+class Test(TestCase):
+    def my_solve(self, testcase_input):
+        sys.stdin = open(testcase_input, 'r')
+        stdout = sys.stdout
+        sys.stdout = open('stdout.txt', 'w')
+        solve()
+        sys.stdout.close()
+        sys.stdout = stdout
+
+    def test_solve(self, testcase_number: str):
+        self.my_solve('test' + testcase_number + '.txt')
+        self.assertEqual(
+            Path('test' + testcase_number + '_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())
+
+    def test1_solve(self):
+        self.test_solve('1')