树状数组

最新推荐文章于 2024-09-02 22:25:23 发布

学术大new牛

最新推荐文章于 2024-09-02 22:25:23 发布

阅读量64

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/qq_36744209/article/details/103690600

版权

数据结构专栏收录该内容

8 篇文章 0 订阅

订阅专栏

#include <iostream>
#include <cstring>
using namespace std;
const int maxn = 10000;
int n,a[maxn],c[maxn],s[maxn];


int lowbit(int i)	//c[i]的区间长度 
{	
	return (-i)&i;
}

void add(int i,int z)	//a[i]加上z
{
	for(;i <= n;i+=lowbit(i)) //直接后继，即父亲i+=lowbit(i)
		c[i] += z;
	
}
int sum(int i)//求前缀和a[1]..a[i] 
{
	int s = 0;
	for(;i>0;i-=lowbit(i))//直接前驱 i-=lowbit(i);
		s+=c[i];
	return s;
} 

int sum(int i,int j)	//求区间和a[i]..a[j]
{
	return sum(j)-sum(i-1);
} 
int main()
{
	memset(c,0,sizeof(c));
	cin >> n;
	for(int i = 1;i <= n;i++)
	{
		cin >> a[i];
		add(i,a[i]);//加入树状数组 
	}
	int x1,x2;
	cin >> x1;
	cout << sum(x1)  << endl;
	cin >> x1 >> x2;
	cout << sum(x1,x2) << endl;
	return 0;
}