백준 11660번 구간 합 구하기 5

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

Author: skysign

백준 11660번 구간 합 구하기 5/main.py

@@ -0,0 +1,29 @@
+import sys
+
+
+def solve():
+    N, M = map(int, sys.stdin.readline().strip().split(' '))
+    maps = [[0 for _ in range(N + 1)]]
+    dp = [[0 for _ in range(N + 1)] for _ in range(N + 1)]
+
+    for _ in range(N):
+        line = list(map(int, sys.stdin.readline().strip().split(' ')))
+        maps.append([0] + line)
+
+    for idx_row in range(1, N + 1):
+        for idx_col in range(1, N + 1):
+            dp[idx_row][idx_col] = dp[idx_row - 1][idx_col] + dp[idx_row][idx_col - 1] - dp[idx_row - 1][idx_col - 1] + maps[idx_row][idx_col]
+
+    for _ in range(M):
+        y1, x1, y2, x2 = map(int, sys.stdin.readline().strip().split(' '))
+        v = map_sum(dp, y1, x1, y2, x2)
+        print(v)
+
+
+def map_sum(dp, y1, x1, y2, x2):
+    rs, rc = y1 - 1, x1 - 1
+    return dp[y2][x2] - (dp[y2][rc] + dp[rs][x2]) + dp[rs][rc]
+
+
+if __name__ == '__main__':
+    solve()

백준 11660번 구간 합 구하기 5/test1.txt

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

백준 11660번 구간 합 구하기 5/test1_answer.txt

@@ -0,0 +1,3 @@
+27
+6
+64

백준 11660번 구간 합 구하기 5/test2.txt

@@ -0,0 +1,7 @@
+2 4
+1 2
+3 4
+1 1 1 1
+1 2 1 2
+2 1 2 1
+2 2 2 2

백준 11660번 구간 합 구하기 5/test2_answer.txt

@@ -0,0 +1,4 @@
+1
+2
+3
+4

백준 11660번 구간 합 구하기 5/test_main.py

@@ -0,0 +1,26 @@
+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')
+
+    def test2_solve(self):
+        self.test_solve('2')