python树状数组模板

最新推荐文章于 2024-04-30 21:03:59 发布

戎梓漩

最新推荐文章于 2024-04-30 21:03:59 发布

阅读量79

点赞数

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

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

版权

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

16 篇文章 0 订阅

订阅专栏

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