백준 11659번 구간 합 구하기 4

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

Author: skysign

백준 11659번 구간 합 구하기 4/main.py

@@ -0,0 +1,20 @@
+import sys
+from typing import List
+
+
+def solve():
+    N, M = map(int, sys.stdin.readline().strip().split(' '))
+    dt: List[int] = [0]
+    dt += map(int, sys.stdin.readline().strip().split(' '))
+    prefixSum: List[int] = [0 for _ in range(len(dt))]
+
+    for i in range(1, len(prefixSum)):
+        prefixSum[i] = prefixSum[i - 1] + dt[i]
+
+    for _ in range(M):
+        fr, to = map(int, sys.stdin.readline().strip().split(' '))
+        print(prefixSum[to] - prefixSum[fr - 1])
+
+
+if __name__ == '__main__':
+    solve()

백준 11659번 구간 합 구하기 4/test1.txt

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

백준 11659번 구간 합 구하기 4/test1_answer.txt

@@ -0,0 +1,3 @@
+12
+9
+1

백준 11659번 구간 합 구하기 4/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')