[Python] BOJ 11660번. 구간 합 구하기 5

작성일 | In |

11660번. 구간 합 구하기 5

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# 11660번. 구간 합 구하기 5


# pypy3로 해야 시간초과 안남
import sys
input = sys.stdin.readline


n, m = map(int, input().split())
box = [[] for i in range(n)]
for i in range(n):
    box[i] = list(map(int, input().split()))

# 열별로 sum
for i in range(n):
    for j in range(n):
        if j == 0:
            pass
        else:
            box[i][j] += box[i][j-1]

for _ in range(m):
    x1, y1, x2, y2 = map(int, input().split())

    ans = 0
    for i in range(x1-1, x2):
        # y2까지 합을 x 범위에서 구하고
        ans += box[i][y2-1]
        # y1 전까지의 합을 빼준다
        if y1 != 1:
            ans -= box[i][y1-2]
    print(ans)

비고