代码拉取完成,页面将自动刷新
# 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)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。