python树状数组模板

最新推荐文章于 2024-09-27 17:31:02 发布

戎梓漩

最新推荐文章于 2024-09-27 17:31:02 发布

阅读量104

点赞数

分类专栏：数据结构与算法文章标签： python 数据结构

本文链接：https://blog.csdn.net/weixin_43817898/article/details/129761525

版权

数据结构与算法专栏收录该内容

16 篇文章 0 订阅

订阅专栏

该文章介绍了一个名为`BinaryIndexedTree`的类，用于处理动态最大值查询。在初始化时，它创建一个大小为n+1的数组c。`update`方法用于更新指定位置x的值，使用了一个位运算技巧。`query`方法则用于查询从1到x的最大值，同样利用位运算优化。这个数据结构常用于解决区间最大值查询问题。

摘要由CSDN通过智能技术生成

class BinaryIndexedTree:
    def __init__(self, n):
        self.n = n
        self.c = [0] * (n + 1)

    def update(self, x, val):
        while x <= self.n:
            self.c[x] = max(self.c[x], val)
            x += x & -x           

    def query(self, x):
        s = 0
        while x:
            s = max(s, self.c[x])
            #x -= x & -x
            x &= (x - 1)
        return s