백준 (boj) 파이썬 - 2167 이차원 배열 합

문제 링크 : https://www.acmicpc.net/problem/2167

2167번: 2차원 배열의 합

 

2차원 배열이기에 2중 for문을 사용하여, 문제내용 그대로 구현했다. 하지만 그냥 python3으로 제출하게되면

 

시간초과가 났기 때문에, 다른 방법을 이용하라는 출제의도같다. 일단은 pypy3로 제출했지만, 다음에 고쳐봐야겠다.

 

 

파이썬 코드 (pypy3로 제출)

import sys
input = sys.stdin.readline
N,M = map(int,input().split())

arr = [];answer = []
for _ in range(N):
    arr.append(list(map(int,input().split())))

K = int(input())
for _ in range(K):
    i,j,x,y = map(int,input().split())
    sum = 0
    for a in range(i-1,x):
        for k in range(j-1,y):
            sum+=arr[a][k]
    answer.append(sum)


for prt in answer :
    print(prt)