树状数组和线段树

简介

查询数组区间和复杂度为log(n)
修改数组值复杂度为log(n)

代码

lowbit函数：最低为第一个1的值

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

创建树状数组

int[] nums; //原数组值
int[] tree; //树状数组
public TreeArr(int[] nums) {
    this.nums = new int[nums.length];
   	this.tree = new int[nums.length+1];
   	for(int i = 0; i < nums.length; i++) {
   		update(i, nums[i]);
           this.nums[i] = nums[i];
   	} 
}

更新数组值

int[] nums; //原数组值
int[] tree; //树状数组
public update(int i, int val) {
    int diff = val - nums[i];
    int x = i + 1;
    while(x < tree.length){
    	tree[x] += diff;
    	x += lowbit(x);
    }
}

求区间和

public int sumRange(int n){
    int x = n + 1;
    int sum = 0;
    while(x > 0){
    	sum += tree[x];
    	x -= lowbit(x);
    }
    return sum;
}
public int sumRange(int i, int j) {
    return sumRange(j) - sumRange(i - 1); //i - 1很重要
}

例题练习

1649. 通过指令创建有序数组

线段树

参考文章

https://zhuanlan.zhihu.com/p/106118909

作用

[1,2,3,4,5]建成的线段树

• 查找区间和O(lgn)
• 查找区间最值O(lgn)
• 更新区间O(lgn)

模版

public static void buildTree(int l, int r, int p, int[] arr, int[] tree) {
    if(l == r) {
        tree[p] = arr[l];
    }
    else {
        int mid = (l + r) / 2;
        buildTree(l, mid, p * 2 + 1, arr, tree); // 先建立左右子节点
        buildTree(mid + 1, r, p * 2 + 2, arr, tree);
        tree[p] = tree[p * 2 + 1] + tree[p * 2 + 2]; // 该节点的值等于左右子节点之和
    }
}
public static void pushDown(int p, int cl, int cr, int[] mark, int[] tree) {
	// 当前索引
    int mid = (cl + cr) / 2;
    // 懒标记下传
    mark[p * 2 + 1] += mark[p];
    mark[p * 2 + 2] += mark[p];
    // 叶子节点的值更新
    tree[p * 2 + 1] += mark[p] * (mid - cl + 1);
    tree[p * 2 + 2] += mark[p] * (cr - mid);
    mark[p] = 0; // 清除标记
}
public static void updateTree(int l, int r, int d, int p, int cl, int cr, int[] mark, int[] tree) {
    // [l, r]:目标区间，需要更新的区间
    // [cl, cr]:当前区间
    if (cl > r || cr < l) { // 区间无交集
        return; // 剪枝
    }
    else if (cl >= l && cr <= r) // 当前节点对应的区间包含在目标区间中
    {
        tree[p] += (cr - cl + 1) * d; // 更新当前区间的值
        if (cr > cl) { // 如果不是叶子节点
            mark[p] += d; // 初始化标记
        }
    }
    else // 与目标区间有交集，但不包含于其中
    {
        int mid = (cl + cr) / 2;
        if(mark[p] != 0) {
            pushDown(p, cl, cr, mark, tree);
        }
        updateTree(l, r, d, p * 2 + 1, cl, mid, mark, tree); // 递归地往下寻找
        updateTree(l, r, d, p * 2 + 2, mid + 1, cr, mark, tree);
        tree[p] = tree[p * 2 + 1] + tree[p * 2 + 2]; // 根据子节点更新当前节点的值
    }
}
public static int queryTree(int l, int r, int p, int cl, int cr, int[] mark, int[] tree) {
    if (cl > r || cr < l)
        return 0;
    else if (cl >= l && cr <= r)
        return tree[p];
    else
    {
        int mid = (cl + cr) / 2;
        pushDown(p, cl, cr, mark, tree);
        return queryTree(l, r, p * 2 + 1, cl, mid, mark, tree) + queryTree(l, r, p * 2 + 2, mid + 1, cr, mark, tree);
    }
}