[BOJ #2167] 2차원 배열의 합

문제

2차원 배열이 주어졌을 때 (i, j) 위치부터 (x, y) 위치까지에 저장되어 있는 수들의 합을 구하는 프로그램을 작성하시오. 배열의 (i, j) 위치는 i행 j열을 나타낸다.

 

입력

첫째 줄에 배열의 크기 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보다 작거나 같다.

 

내가 쓴 코드

from sys import stdin

N, M = map(int, stdin.readline().split())      # N x M 크기의 2차원 배열
arr = [list(map(int, stdin.readline().split())) for _ in range(N)]
K = int(stdin.readline())
for k in range(K):
    i1, j1, i2, j2 = map(int, stdin.readline().split())

    arr_sum = 0
    for i in range(i1-1, i2):
        for j in range(j1-1, j2):
            arr_sum += arr[i][j]

    print(arr_sum)

 

출력 결과

 

제출 결과

 

파이참으로 코드 작성하고 실행해보고 있는데, 출력 형식이 이상하다.. 다른 분에게 물어봤더니, 파이참이 원래 출력이 이상하다고... vscode로 실행해보면 제대로 나올거라고 해주셨다. 나 혼자 출력 형식 바꾸느라 애먹고있었음^^ 하지만 귀찮아서 그냥 파이참으로 계속 할겁니다! 하하호호 :-)