ps:problems:boj:16507
어두운 건 무서워
ps | |
---|---|
링크 | acmicpc.net/… |
출처 | BOJ |
문제 번호 | 16507 |
문제명 | 어두운 건 무서워 |
레벨 | 실버 1 |
분류 |
누적합 |
시간복잡도 | O(r*c + q) |
인풋사이즈 | r<=1000, c<=1000, q<=10000 |
사용한 언어 | Python |
제출기록 | 108860KB / 472ms |
최고기록 | 472ms |
해결날짜 | 2022/05/30 |
태그 |
풀이
- 2차원 구간의 평균을 구하는 것은, 단순하게 2차원 구간합을 구하고 크기로 나눠주면 된다.
- 2차원 구간합은 2차원 누적합을 이용해서 O(1)에 구할수 있다. 구간 합 구하기 5 참고.
- 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()
ps/problems/boj/16507.txt · 마지막으로 수정됨: 2022/05/31 15:26 저자 teferi
토론