如何实现数状数组？

最新推荐文章于 2024-09-06 10:38:06 发布

Durian_piza

最新推荐文章于 2024-09-06 10:38:06 发布

阅读量68

点赞数

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

本文链接：https://blog.csdn.net/Durian_piza/article/details/129985244

版权

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

7 篇文章 0 订阅

订阅专栏

该文章介绍了Java中实现的数状数组类(classNumArray)，包括lowbit函数用于计算低位异或，用于查询(query)和添加(add)操作。类提供了初始化、更新(update)数值以及求区间和(sumRange)的功能，适用于高效的数据处理。

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

数状数组

一、数状数组

一、数状数组

在这里插入图片描述

class NumArray {
    int[] tree;
    int lowbit(int x) {
        return x & -x;
    }
    int query(int x) {
        int ans = 0;
        for (int i = x; i > 0; i -= lowbit(i)) ans += tree[i];
        return ans;
    }
    void add(int x, int u) {
        for (int i = x; i <= n; i += lowbit(i)) tree[i] += u;
    }

    int[] nums;
    int n;
    public NumArray(int[] _nums) {
        nums = _nums;
        n = nums.length;
        tree = new int[n + 1];
        for (int i = 0; i < n; i++) add(i + 1, nums[i]);
    }
    
    public void update(int i, int val) {
        add(i + 1, val - nums[i]);
        nums[i] = val;
    }
    
    public int sumRange(int l, int r) {
        return query(r + 1) - query(l);
    }
}