代码拉取完成,页面将自动刷新
# Time: ctor: O(n),
# update: O(logn),
# query: O(logn)
# Space: O(n)
class NumArray(object):
def __init__(self, nums):
"""
initialize your data structure here.
:type nums: List[int]
"""
if not nums:
return
self.__nums = nums
self.__bit = [0] * (len(self.__nums) + 1)
for i in xrange(1, len(self.__bit)):
self.__bit[i] = nums[i-1] + self.__bit[i-1]
for i in reversed(xrange(1, len(self.__bit))):
last_i = i - (i & -i)
self.__bit[i] -= self.__bit[last_i]
def update(self, i, val):
"""
:type i: int
:type val: int
:rtype: int
"""
if val - self.__nums[i]:
self.__add(i, val - self.__nums[i])
self.__nums[i] = val
def sumRange(self, i, j):
"""
sum of elements nums[i..j], inclusive.
:type i: int
:type j: int
:rtype: int
"""
return self.__sum(j) - self.__sum(i-1)
def __sum(self, i):
i += 1
ret = 0
while i > 0:
ret += self.__bit[i]
i -= (i & -i)
return ret
def __add(self, i, val):
i += 1
while i <= len(self.__nums):
self.__bit[i] += val
i += (i & -i)
# Time: ctor: O(n),
# update: O(logn),
# query: O(logn)
# Space: O(n)
# Segment Tree solution.
class NumArray2(object):
def __init__(self, nums,
query_fn=lambda x, y: x+y,
update_fn=lambda x, y: y,
default_val=0):
"""
initialize your data structure here.
:type nums: List[int]
"""
N = len(nums)
self.__original_length = N
self.__tree_length = 2**(N.bit_length() + (N&(N-1) != 0))-1
self.__query_fn = query_fn
self.__update_fn = update_fn
self.__default_val = default_val
self.__tree = [default_val for _ in range(self.__tree_length)]
self.__lazy = [None for _ in range(self.__tree_length)]
self.__constructTree(nums, 0, self.__original_length-1, 0)
def update(self, i, val):
self.__updateTree(val, i, i, 0, self.__original_length-1, 0)
def sumRange(self, i, j):
return self.__queryRange(i, j, 0, self.__original_length-1, 0)
def __constructTree(self, nums, left, right, idx):
if left > right:
return
if left == right:
self.__tree[idx] = self.__update_fn(self.__tree[idx], nums[left])
return
mid = left + (right-left)//2
self.__constructTree(nums, left, mid, idx*2 + 1)
self.__constructTree(nums, mid+1, right, idx*2 + 2)
self.__tree[idx] = self.__query_fn(self.__tree[idx*2 + 1], self.__tree[idx*2 + 2])
def __apply(self, left, right, idx, val):
self.__tree[idx] = self.__update_fn(self.__tree[idx], val)
if left != right:
self.__lazy[idx*2 + 1] = self.__update_fn(self.__lazy[idx*2 + 1], val)
self.__lazy[idx*2 + 2] = self.__update_fn(self.__lazy[idx*2 + 2], val)
def __updateTree(self, val, range_left, range_right, left, right, idx):
if left > right:
return
if self.__lazy[idx] is not None:
self.__apply(left, right, idx, self.__lazy[idx])
self.__lazy[idx] = None
if range_left > right or range_right < left:
return
if range_left <= left and right <= range_right:
self.__apply(left, right, idx, val)
return
mid = left + (right-left)//2
self.__updateTree(val, range_left, range_right, left, mid, idx*2 + 1)
self.__updateTree(val, range_left, range_right, mid+1, right, idx*2 + 2)
self.__tree[idx] = self.__query_fn(self.__tree[idx*2 + 1],
self.__tree[idx*2 + 2])
def __queryRange(self, range_left, range_right, left, right, idx):
if left > right:
return self.__default_val
if self.__lazy[idx] is not None:
self.__apply(left, right, idx, self.__lazy[idx])
self.__lazy[idx] = None
if right < range_left or left > range_right:
return self.__default_val
if range_left <= left and right <= range_right:
return self.__tree[idx]
mid = left + (right-left)//2
return self.__query_fn(self.__queryRange(range_left, range_right, left, mid, idx*2 + 1),
self.__queryRange(range_left, range_right, mid + 1, right, idx*2 + 2))
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。