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

Python实现线段树

Java 更新时间:发布时间: 百科书网 趣学号
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.051e.com
本文地址:http://www.051e.com/it/986500.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号