BAEKJOON_2167 "2차원 배열의 합" PYTHON

[백준] 2167번 Python 파이썬

 

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

2167번: 2차원 배열의 합

 

11660 문제와 유사하여 행렬의 크기 등만 조금 수정하였다.

아래 사이트 참고!

 

https://rilla0511.tistory.com/95

BAEKJOON_11660 "구간 합 구하기 5" PYTHON

import sys

n, m = map(int,sys.stdin.readline().split())
tb = [[0 for j in range(m)] for i in range(n)]

for i in range(n):
    lt = list(map(int, sys.stdin.readline().split()))
    for j in range(m):
        tb[i][j] = lt[j]
        
st = [[0 for j in range(m)] for i in range(n)]

for i in range(n):
    for j in range(m):
        if i == 0 and j == 0:
            st[i][j] = tb[i][j]
        elif i == 0:
            st[i][j] = st[i][j-1] + tb[i][j]
        elif j == 0:
            st[i][j] = st[i-1][j] + tb[i][j]
        else:
            st[i][j] = st[i-1][j] + st[i][j-1] - st[i-1][j-1] + tb[i][j]

answer = []

for i in range(int(sys.stdin.readline())):
    x1, y1, x2, y2 = map(int, sys.stdin.readline().split())
    if (x1 == x2 and y1 == y2):
        answer.append(tb[x2-1][y2-1])
    elif x1 == 1 and y1 == 1:
        answer.append(st[x2-1][y2-1])
    elif x1 == 1:
        answer.append(st[x2-1][y2-1] - st[x2-1][y1-2])
    elif y1 == 1:
        answer.append(st[x2-1][y2-1] - st[x1-2][y2-1])
    else:
        answer.append(st[x2-1][y2-1] - st[x1-2][y2-1] - st[x2-1][y1-2] + st[x1-2][y1-2])

for i in answer:
    print(i)

 

혼자 공부하며 올리는 블로그입니다. 틀린 내용은 댓글 남겨주시면 감사하겠습니다.