栏目分类:
子分类:
返回
文库吧用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
文库吧 > IT > 软件开发 > 后端开发 > C/C++/C#

C++ 树状数组

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

C++ 树状数组

1.作用:

 快速单点修改和前缀和查询,复杂度为O(logn)

2.特点:

每个树节点的节点值是所管理的子节点值之和,lowbit()通过取最后一个二进制位,配合与原索引的加 / 减操作进行爬树 / 下树,爬树过程中更新元素增量,下树过程中累计前缀和,由于是树结构,所以更新和获取操作的时间复杂度为O(logn)

3.示意图:

4.C++代码:
#include
#include
using namespace std;



class BIT {
private:
	int lowbit(const int& x) { // 树状数组的基本操作,可以理解为爬树和下树的过程
		return x & (-x);
	}
	vectortree; //树状数组
	int sizeT; // 数组大小
public:
	// 因为lowbit(0)还是0,没有意义,因此tree[0]是不用的
	// tree[0]是不用的,从tree[1]开始,因此tree(size+1)
	explicit BIT(int size):sizeT(size), tree(size+1) {} // explicit防止类构造函数的隐式自动转换

	void update(int idx, int val) { // 元素更新,val为元素的增量,如果原数组值为0,则直接输入更新的元素值
		while (idx <= sizeT) {
			tree[idx] += val; // 爬树的过程中更新增量
			idx += lowbit(idx); // 爬树
		}
	}

	int query(int idx) { // 获得前缀和
		int prefix = 0;
		while (idx > 0) { // tree[0]是不用的!
			prefix += tree[idx]; // 下树的过程中累计前缀和
			idx -= lowbit(idx); // 下树
		}
		return prefix;
	}

	void print(vector& arr) {
		for (const int& num : arr) cout << num << ' ';
		cout << endl;
	}
};

int main() {
	vectorarr(10);
	BIT bit(10);
	for (int i = 0; i < 10; ++i) {
		arr[i] = rand() % 20;
		bit.update(i+1, arr[i]);
	}
	bit.print(arr);
	for (int i = 0; i < 10; ++i) {
		cout << bit.query(i+1) << ' ';
	}
	cout << endl;
	return 0;
}

 做笔记使用,树状数组的理解方面还不是特别透彻,如果有大神看到了还希望多多指点!!!

谢谢!!溜了溜了~~~

转载请注明:文章转载自 www.wk8.com.cn
本文地址:https://www.wk8.com.cn/it/1038299.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 wk8.com.cn

ICP备案号:晋ICP备2021003244-6号