树状数组-Java代码纯享版

/**
 原始数组下标从0开始,
 bitTree数组下标从1开始,目的是为了符合lowbit操作,lowbit可以理解为构造新数组的一种方法(比较直观简单的理解)
**/
public class BITDemo {
    int[] nums;
    int[] bitree;
    int n;

    public BITDemo(int[] _nums){
        this.nums = _nums;
        this.n = nums.length;
        bitree = new int[this.n+1];
        // 构造出bitTree
        for(int i=0;i<n;i++){
            add(i+1, nums[i]);
        }
    }

    public int lowBit(int x){
        return x&(-x);
    }

    // 单点更新,新增值!将新增的值累加到父节点上
    public void add(int index, int increase){
        for(int i=index;i<=n;i+=lowBit(i)){
            bitree[i]+=increase;
        }
    }

    // 单点修改,将某个位置的值修改为 value
    public void update(int index, int value){
        // 核心就是利用add来增加差值
        add(index, value-nums[index-1]);
        nums[index-1] = value;
    }

    // 查询前缀和
    public int query(int index){
        int res = 0;
        for(int i=index;i>0;i-=lowBit(i)){
            res+=bitree[i];
        }
        return res;
    }

    // 查询区间和
    public int range(int left, int right){
        return query(right)-query(left);
    }

    public static void main(String[] args) {
        // 所有的操作都是打在bitTree 上
        int[] nums = new int[]{2,3,5,1,4,2,1};
        BITDemo bitDemo = new BITDemo(nums);
        System.out.println(bitDemo.query(2));   //5
        System.out.println(bitDemo.query(5));   //15
        System.out.println(bitDemo.range(2, 5)); //10
        bitDemo.update(1,1);    // 1 将1位置重置为1
        System.out.println(bitDemo.query(1));//1
        bitDemo.add(1,10);  // 将1位置加10
        System.out.println(bitDemo.query(1));   //11
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
树状数组(Fenwick Tree)是一种用于快速维护数组前缀和的数据结构。它可以在 $O(\log n)$ 的时间内完成单点修改和前缀查询操作,比线段树更加简洁高效。 下面是 Java 实现的树状数组详解: 首先,在 Java 中我们需要使用数组来表示树状数组,如下: ``` int[] tree; ``` 接着,我们需要实现两个基本操作:单点修改和前缀查询。 单点修改的实现如下: ``` void update(int index, int value) { while (index < tree.length) { tree[index] += value; index += index & -index; } } ``` 该函数的参数 `index` 表示要修改的位置,`value` 表示修改的值。在函数内部,我们使用了一个 `while` 循环不断向上更新树状数组中相应的节点,直到到达根节点为止。具体来说,我们首先将 `tree[index]` 加上 `value`,然后将 `index` 加上其最后一位为 1 的二进制数,这样就可以更新其父节点了。例如,当 `index` 为 6 时,其二进制表示为 110,最后一位为 2^1,加上后变为 111,即 7,这样就可以更新节点 7 了。 前缀查询的实现如下: ``` int query(int index) { int sum = 0; while (index > 0) { sum += tree[index]; index -= index & -index; } return sum; } ``` 该函数的参数 `index` 表示要查询的前缀的结束位置,即查询 $[1, index]$ 的和。在函数内部,我们同样使用了一个 `while` 循环不断向前查询树状数组中相应的节点,直到到达 0 为止。具体来说,我们首先将 `sum` 加上 `tree[index]`,然后将 `index` 减去其最后一位为 1 的二进制数,这样就可以查询其前一个节点了。例如,当 `index` 为 6 时,其二进制表示为 110,最后一位为 2^1,减去后变为 100,即 4,这样就可以查询节点 4 的值了。 最后,我们还需要初始化树状数组,将其全部置为 0。初始化的实现如下: ``` void init(int[] nums) { tree = new int[nums.length + 1]; for (int i = 1; i <= nums.length; i++) { update(i, nums[i - 1]); } } ``` 该函数的参数 `nums` 表示初始数组的值。在函数内部,我们首先创建一个长度为 `nums.length + 1` 的数组 `tree`,然后逐个将 `nums` 中的元素插入到树状数组中。具体来说,我们调用 `update(i, nums[i - 1])` 来将 `nums[i - 1]` 插入到树状数组的第 `i` 个位置。 到此为止,我们就完成了树状数组的实现。可以看到,树状数组代码比线段树要简洁很多,而且效率也更高。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小吴同学GOGOGO

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值