栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

Python实现线段树

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Python实现线段树

class SegmentTreeDynamic:
    class Node:
        left, right = None, None
        val, add = 0, 0

        # def __init__(self, left, right, val, add):
        #     self.left = left
        #     self.right = right
        #     self.val = val
        #     self.add = add
        def __init__(self):
            pass

    N = int(1e9)
    root = Node()

    def update(self, node, start, end, l, r, val):
        if l <= start and end <= r:
            node.val += ((end - start) + 1) * val
            node.add = val
            return
        mid = (start + end) >> 1
        self.pushDown(node, mid - start + 1, end - mid)
        if l <= mid: self.update(node.left, start, mid, l, r, val)
        if r > mid: self.update(node.right, mid + 1, end, l, r, val)
        self.pushup(node)

    def pushup(self, node):
        node.val = node.left.val + node.right.val

    def pushDown(self, node, leftNum, rightNum):
        if node.left is None: node.left = self.Node()
        if node.right is None: node.right = self.Node()
        if node.add == 0: return
        node.left.val = node.add * leftNum
        node.right.val = node.add * rightNum
        node.add = 0

    def query(self, node, start, end, l, r):
        if l <= start and end <= r: return node.val
        mid, ans = (start + end) >> 1, 0
        self.pushDown(node, mid - start + 1, end - mid)
        if l <= mid: ans += self.query(node.left, start, mid, l, r)
        if r > mid: ans += self.query(node, mid + 1, end, l, r)
        return ans

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/986500.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号