Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
range-sum-query-2d-mutable.py 2.33 KB
一键复制 编辑 原始数据 按行查看 历史
Allen Liu 提交于 2018-10-13 01:56 +08:00 . add complexity
# Time: ctor: O(m * n)
# update: O(logm * logn)
# query: O(logm * logn)
# Space: O(m * n)
class NumMatrix(object):
def __init__(self, matrix):
"""
initialize your data structure here.
:type matrix: List[List[int]]
"""
if not matrix:
return
self.__matrix = matrix
self.__bit = [[0] * (len(self.__matrix[0]) + 1) \
for _ in xrange(len(self.__matrix) + 1)]
for i in xrange(1, len(self.__bit)):
for j in xrange(1, len(self.__bit[0])):
self.__bit[i][j] = matrix[i-1][j-1] + self.__bit[i-1][j] + \
self.__bit[i][j-1] - self.__bit[i-1][j-1]
for i in reversed(xrange(1, len(self.__bit))):
for j in reversed(xrange(1, len(self.__bit[0]))):
last_i, last_j = i - (i & -i), j - (j & -j)
self.__bit[i][j] = self.__bit[i][j] - self.__bit[i][last_j] - \
self.__bit[last_i][j] + self.__bit[last_i][last_j]
def update(self, row, col, val):
"""
update the element at matrix[row,col] to val.
:type row: int
:type col: int
:type val: int
:rtype: void
"""
if val - self.__matrix[row][col]:
self.__add(row, col, val - self.__matrix[row][col])
self.__matrix[row][col] = val
def sumRegion(self, row1, col1, row2, col2):
"""
sum of elements matrix[(row1,col1)..(row2,col2)], inclusive.
:type row1: int
:type col1: int
:type row2: int
:type col2: int
:rtype: int
"""
return self.__sum(row2, col2) - self.__sum(row2, col1 - 1) - \
self.__sum(row1 - 1, col2) + self.__sum(row1 - 1, col1 - 1)
def __sum(self, row, col):
row += 1
col += 1
ret = 0
i = row
while i > 0:
j = col
while j > 0:
ret += self.__bit[i][j]
j -= (j & -j)
i -= (i & -i)
return ret
def __add(self, row, col, val):
row += 1
col += 1
i = row
while i <= len(self.__matrix):
j = col
while j <= len(self.__matrix[0]):
self.__bit[i][j] += val
j += (j & -j)
i += (i & -i)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助