LeetCode：304. 二維區域和檢索 - 矩陣不可變————中等

目錄

• • 題目
  • Code
  • 運行結果

題目

給定一個二維矩陣，計算其子矩形範圍內元素的總和，該子矩陣的左上角爲 (row1, col1) ，右下角爲 (row2, col2) 。

上圖子矩陣左上角 (row1, col1) = (2, 1) ，右下角(row2, col2) = (4, 3)，該子矩形內元素的總和爲 8。

示例：
給定 matrix = [
  [3, 0, 1, 4, 2],
  [5, 6, 3, 2, 1],
  [1, 2, 0, 1, 5],
  [4, 1, 0, 1, 7],
  [1, 0, 3, 0, 5]
]

sumRegion(2, 1, 4, 3) -> 8
sumRegion(1, 1, 2, 2) -> 11
sumRegion(1, 2, 2, 4) -> 12
 

提示：
你能夠假設矩陣不可變。
會屢次調用 sumRegion 方法。
你能夠假設 row1 ≤ row2 且 col1 ≤ col2 。

---

Code

class NumMatrix:

    def __init__(self, matrix: List[List[int]]):
        if not matrix or not matrix[0]:
            M, N = 0, 0
        else:
            M, N = len(matrix), len(matrix[0])
        self.preSum = [[0] * (N + 1) for _ in range(M + 1)]
        for i in range(M):
            for j in range(N):
                self.preSum[i + 1][j + 1] = self.preSum[i][j + 1] + self.preSum[i + 1][j]  - self.preSum[i][j] + matrix[i][j]


    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
        return self.preSum[row2 + 1][col2 + 1] - self.preSum[row2 + 1][col1] - self.preSum[row1][col2 + 1] + self.preSum[row1][col1]

---

運行結果

本文同步分享在 博客「Kinght_123」（CSDN）。
若有侵權，請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」，歡迎正在閱讀的你也加入，一塊兒分享。