树状数组

最新推荐文章于 2025-04-24 15:39:27 发布

weixin_30655219

最新推荐文章于 2025-04-24 15:39:27 发布

阅读量42

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Aragaki/p/7226766.html

本文介绍了一种使用树状数组实现区间求和的方法。通过更新树状数组中对应节点的值来完成对原数组的修改，并能快速计算任意区间的元素之和。适用于解决动态区间加法/查询问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
using namespace std;
int n,m,i,num[100001],t[200001],l,r;//num:原数组；t：树状数组 
int lowbit(int x) 
{
    return x&(-x); //右起第一个1的位置为第k个 返回值则为2^(k-1)
}
void update(int x,int p)//更新第x 的值
{
    while(x<=n)
    {
        t[x]+=p;
        x+=lowbit(x);//不断找到自己的祖先 更新值
    }
    return;
}
int sum(int k)//前k个数的和 
{
    int ans=0;
    while(k>0)
    {
        ans+=t[k];
        k-=lowbit(k);
    }
    return ans;
}
int ask(int l,int r)//求l-r区间和 
{
    return sum(r)-sum(l-1); 
}
int main()
{
    cin>>n>>m;
    for(i=1;i<=n;i++)
    {
        cin>>num[i];
        update(i,num[i]);
    }
    for(i=1;i<=m;i++)
    {
        cin>>l>>r;
        cout<<ask(l,r)<<endl;
    }
    return 0;
}