황소개발자

백준 2167 파이썬 python : 2차원 배열의 합 @@황소처럼 우직하게@@ 김정은은 살아있는 것일까.. 본문

백준 문제 풀이

백준 2167 파이썬 python : 2차원 배열의 합 @@황소처럼 우직하게@@ 김정은은 살아있는 것일까..

hjp845 2020. 4. 25. 14:32
반응형

지금 시국.. 김정은 사망설 난무..

북한 입장 아직까지도 안나오는거 보면, 뭔가 있긴 한갑다..

dp 배열의 정의는

dp[i][j]  :  1행 1열 부터 i 행 j열 까지의 합

답 구할 땐, dp[y][x] - dp[y][j - 1] - dp[i - 1][x] + dp[i - 1][j - 1] 

이런식으로 빼주고 두번 빼진건 한 번 더해주고

import sys
input = sys.stdin.readline

h, w = map(int, input().split())
mat = []
for i in range(h):
    mat.append(list(map(int, input().split())))

dp = [[0 for _ in range(w + 1)] for _ in range(h + 1)]
for i in range(1, h + 1):
    for j in range(1, w + 1):
        dp[i][j] = mat[i - 1][j - 1] + dp[i - 1][j] + dp[i][j - 1] - dp[i - 1][j - 1]

k = int(input())
for _ in range(k):
    i, j, y, x = map(int, input().split())
    ans = dp[y][x] - dp[y][j - 1] - dp[i - 1][x] + dp[i - 1][j - 1]
    print(ans)
반응형
Comments