[백준/Python] 16507번 : 어두운 건 무서워

문제

https://www.acmicpc.net/problem/16507

16507번: 어두운 건 무서워

---

풀이

누적 합을 저장하기 위해 새로운 2차원 배열(ps)을 만들어 준다.

이중 반복문을 이용하여 누적합을 저장한다.
ps[i][j] = ps[i][j-1] + ps[i-1][j] - ps[i-1][j-1] + li[i-1][j-1]

입력받은 좌표에 해당하는 밝기의 합을 구한다.
ps[x2][y2] - ps[x2][y1-1] - ps[x1-1][y2] + ps[x1-1][y1-1]

픽셀 수로 나누어 평균을 구한다.
(x2-x1+1)*(y2-y1+1)

---

코드

import sys
input = sys.stdin.readline

r, c, q = map(int, input().split())
li = [[*map(int, input().split())] for _ in range(r)]

ps = [[0]*(c+1) for _ in range(r+1)]
for i in range(1, r+1):
    for j in range(1, c+1):
        ps[i][j] = ps[i][j-1] + ps[i-1][j] - ps[i-1][j-1] + li[i-1][j-1]

for _ in range(q):
    x1, y1, x2, y2 = map(int, input().split())
    print((ps[x2][y2] - ps[x2][y1-1] - ps[x1-1][y2] + ps[x1-1][y1-1]) // ((x2-x1+1)*(y2-y1+1)))