728x90
문제
2차원 배열이 주어졌을 때 (i, j) 위치부터 (x, y) 위치까지에 저장되어 있는 수들의 합을 구하는 프로그램을 작성하시오.
배열의 (i, j) 위치는 i행 j열을 나타낸다.
https://www.acmicpc.net/problem/2167
입력
첫째 줄에 배열의 크기 N, M(1 ≤ N, M ≤ 300)이 주어진다.
다음 N개의 줄에는 M개의 정수로 배열이 주어진다.
배열에 포함되어 있는 수는 절댓값이 10,000보다 작거나 같은 정수이다.
그 다음 줄에는 합을 구할 부분의 개수 K(1 ≤ K ≤ 10,000)가 주어진다.
다음 K개의 줄에는 네 개의 정수로 i, j, x, y가 주어진다(1 ≤ i ≤ x ≤ N, 1 ≤ j ≤ y ≤ M).
출력
K개의 줄에 순서대로 배열의 합을 출력한다. 배열의 합은 231-1보다 작거나 같다.
예제
# input
2 3
1 2 4
8 16 32
3
1 1 2 3
1 2 1 2
1 3 2 3
# output
63
2
36문제 풀이
import sys
input = sys.stdin.readline
# 입력 받기
N, M = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(N)]
# DP를 위한 배열 초기화
dp = [[0] * (M+1) for _ in range(N+1)]
# arr 누적합 구하여 dp 배열에 저장
# arr[i-1][j-1]: 현재 위치의 원본 배열 값
# dp[i][j-1]: 현재 위치의 왼쪽 칸까지의 누적 합
# dp[i-1][j]: 현재 위치의 위쪽 칸까지의 누적 합
# dp[i-1][j-1]: 현재 위치의 왼쪽 위 대각선 칸까지의 누적 합. 중복된 부분을 빼기 위함
for i in range(1, N+1):
for j in range(1, M+1):
dp[i][j] = arr[i-1][j-1] + dp[i][j-1] + dp[i-1][j] - dp[i-1][j-1]
K = int(input())
for _ in range(K):
i, j, x, y = map(int, input().split())
print(dp[x][y] - dp[x][j-1] - dp[i-1][y] + dp[i-1][j-1])파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.
728x90
