====== 어두운 건 무서워 ====== ===== 풀이 ===== * 2차원 구간의 평균을 구하는 것은, 단순하게 2차원 구간합을 구하고 크기로 나눠주면 된다. * 2차원 구간합은 2차원 누적합을 이용해서 O(1)에 구할수 있다. [[ps:problems:boj:11660]] 참고. * 2차원 누적합을 O(R*C)에 만들고, Q개의 쿼리를 각각 O(1)에 처리하므로 총 시간복잡도는 O(R*C + Q) ===== 코드 ===== """Solution code for "BOJ 16507. 어두운 건 무서워". - Problem link: https://www.acmicpc.net/problem/16507 - Solution link: http://www.teferi.net/ps/problems/boj/16507 Tags: [Prefix sum] """ import sys def create_2d_prefix_sum(nums): prefix_sum = [[0] * (len(nums[0]) + 1)] for row in nums: prefix_sum.append([ps := 0] + [(ps := ps + num) + p for num, p in zip(row, prefix_sum[-1][1:])]) return prefix_sum def main(): # pylint: disable=unused-variable R, C, Q = [int(x) for x in sys.stdin.readline().split()] nums = [[int(x) for x in sys.stdin.readline().split()] for _ in range(R)] prefix_sum = create_2d_prefix_sum(nums) for _ in range(Q): r1, c1, r2, c2 = [int(x) for x in sys.stdin.readline().split()] pixel_sum = (prefix_sum[r2][c2] - prefix_sum[r1 - 1][c2] - prefix_sum[r2][c1 - 1] + prefix_sum[r1 - 1][c1 - 1]) size = (r2 - r1 + 1) * (c2 - c1 + 1) print(pixel_sum // size) if __name__ == '__main__': main() {{tag>BOJ ps:problems:boj:실버_1}}