[파이썬, Python] 백준 16507번 : 어두운 건 무서워

백준 16507번 : 어두운 건 무서워

(문제 바로가기)

 

 

내 코드

import sys

input = sys.stdin.readline
r, c, q = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(r)]
sum_arr = [[0] * (c + 1) for _ in range(r + 1)]

for i in range(1, r + 1):
    for j in range(1, c + 1):
        sum_arr[i][j] = sum_arr[i - 1][j] + sum_arr[i][j - 1] + arr[i - 1][j - 1] - sum_arr[i - 1][j - 1]

for _ in range(q):
    r1, c1, r2, c2 = map(int, input().split())
    total = sum_arr[r2][c2] - sum_arr[r1 - 1][c2] - sum_arr[r2][c1 - 1] + sum_arr[r1 - 1][c1 - 1]
    print(total // ((r2 - r1 + 1) * (c2 - c1 + 1)))