알고리즘 문제 풀이

3 / 16

백준 2167 2차원 배열의 합 with Python

알고리즘

문제 링크


알고리즘

  • 구현

풀이 과정

파이썬의 리스트 슬라이싱을 사용하면 속도가 빠르기 때문에 시간초과가 나지 않는다.

전체 코드

import sys

input = sys.stdin.readline

N, M = map(int, input().split())

lst = [list(map(int, input().split())) for _ in range(N)]

for _ in range(int(input())):
    y1, x1, y2, x2 = map(int, input().split())
    print(sum([sum(lst[i][x1 - 1 : x2]) for i in range(y1 - 1, y2)]))