常州网站建设价格,沉默是金吉他谱,网络营销方案例文,120平方装修预算表题目截图 题目分析
关键就是记录每次操作2时#xff0c;nums1中的1的个数这就需要实现线段树进行区间反转以及区间求和
ac code
class Solution:def handleQuery(self, nums1: List[int], nums2: List[int], queries: List[List[int]]) - List[int]:n len(nums1)m le…题目截图 题目分析
关键就是记录每次操作2时nums1中的1的个数这就需要实现线段树进行区间反转以及区间求和
ac code
class Solution:def handleQuery(self, nums1: List[int], nums2: List[int], queries: List[List[int]]) - List[int]:n len(nums1)m len(queries)seg_tree SegTree(nums1)# 只需要记录每次2操作时nums1中有多少个1即可total sum(nums2)ans []for i in range(m):if queries[i][0] 1:l queries[i][1]r queries[i][2]seg_tree.reverse_range(l, r)elif queries[i][0] 2:total seg_tree.sum_range(0, n - 1) * queries[i][1]elif queries[i][0] 3:ans.append(total)return ansclass SegTree:def __init__(self, nums):n len(nums)self.arr [SegNode() for _ in range(n * 4 1)]self.build(1, 0, n - 1, nums)def sum_range(self, left, right):return self.query(1, left, right)def reverse_range(self, left, right):self.modify(1, left, right)def build(self, id, l, r, nums):arr self.arrarr[id] SegNode()arr[id].l larr[id].r rarr[id].lazytag Falseif l r:arr[id].sum nums[l]returnmid (l r) 1self.build(2 * id, l, mid, nums)self.build(2 * id 1, mid 1, r, nums)arr[id].sum arr[2 * id].sum arr[2 * id 1].sum# pushdown函数下传懒标记即将当前区间的修改情况下传到其左右孩子结点def pushdown(self, x):arr self.arrif arr[x].lazytag:arr[2 * x].lazytag not arr[2 * x].lazytagarr[2 * x].sum arr[2 * x].r - arr[2 * x].l 1 - arr[2 * x].sumarr[2 * x 1].lazytag not arr[2 * x 1].lazytagarr[2 * x 1].sum arr[2 * x 1].r - arr[2 * x 1].l 1 - arr[2 * x 1].sumarr[x].lazytag False# 区间修改def modify(self, id, l, r):arr self.arrif arr[id].l l and arr[id].r r:arr[id].sum (arr[id].r - arr[id].l 1) - arr[id].sumarr[id].lazytag not arr[id].lazytagreturnself.pushdown(id)mid (arr[id].l arr[id].r) 1if arr[2 * id].r l:self.modify(2 * id, l, r)if arr[2 * id 1].l r:self.modify(2 * id 1, l, r)arr[id].sum arr[2 * id].sum arr[2 * id 1].sum# 区间查询def query(self, id, l, r):arr self.arrif arr[id].l l and arr[id].r r:return arr[id].sumif arr[id].r l or arr[id].l r:return 0self.pushdown(id)mid (arr[id].l arr[id].r) 1res 0if arr[2 * id].r l:res self.query(2 * id, l, r)if arr[2 * id 1].l r:res self.query(2 * id 1, l, r)return resclass SegNode:def __init__(self):self.l 0self.r 0self.sum 0self.lazytag False
