树桩数组

李昌晋

于 2024-08-10 00:49:35 发布

阅读量492

点赞数 5

分类专栏：数据结构与算法文章标签： java 开发语言算法数据结构

本文链接：https://blog.csdn.net/weixin_55245206/article/details/141065731

版权

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

16 篇文章 0 订阅

订阅专栏

什么是树状数组

图片描述

可以解决大部分基于区间上的更新以及求和问题。

$l o w bi t$ ( $x$ & $- x$ )

解释一：

$x$ 的二进制最右边第一个出现的 $1$ 和后面出现的 $0$ 组成的 $10$ 进制数：

$l o w bi t (6) = 110 - > 2$

$l o w bi t (8) = 1000 - > 8$

解释二：

$x$ 的二进制最右边有 $k$ 个连续的 $0$ ， $lowbit(x) = 2^x$ ;

树状数组实现

public class TreeArray{
    int[] A;//原始数据
    int[] C;//树状数组
    int n;//数组元素个数
    public TreeArray(int[] A){
        this.A = new int[A.length];
        n = A.length;
        C = new int[n+1];
        for(int i=1;i<=A.length;i++){
            updata(i,A[i-1]);
        }
    }
    public int lowbit(int x){
        return x & -x;
    }
    public void updata(int i,int val){
        //原来的值A[i-1]，现在的值是val
        int data = val - A[i-1];//新旧值之间的差距
        A[i-1] = val;
        for(int pos = i;pos<=n;pos = pos+lowbit(pos)){
            C[pos] += data;
        }
    }
    public int sum(int pos){
        int sum = 0;
        for(int i=pos;i>=1;i-=lowbit(i)){
            sum = sum + C[i];
            return sum;
        }
    }
    public int sumRange(int start,int end){
        if(start<1 || start>n || end<1 || end>n) return -1;
        else
            return sum(end) - sum(start-0);
    }
}