树状数组与线段树模板

题目链接：1264. 动态求连续区间和 - AcWing题库

树状数组

AC代码：

#include<iostream>
using namespace std;

const int MAXN = 100005;
int m,n;
int k,a,b;
int temp;
int w[MAXN],tr[MAXN];

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

void add(int x,int v)
{
    for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=v;
}

int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];

    return res;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i)
    {
        cin>>temp;
        add(i,temp);
    }
    for(int i=1;i<=m;++i)
    {
        cin>>k>>a>>b;
        if(k==1) add(a,b);
        else if(k==0) cout<<query(b)-query(a-1)<<endl;
    }

    return 0;
}

线段树模板

#include<iostream>
using namespace std;

const int MAXN = 100005;

int n,m;
int w[MAXN];
int k,a,b;
struct Node
{
    int l,r;
    int sum;
}tr[MAXN*4];

//计算结点u的sum值
void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1 | 1].sum;
}

//建立编号为u的结点
void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[l]};
    else
    {
        tr[u]={l,r};
        int mid = l+r >> 1;
        build(u<<1,l,mid);
        build(u<<1 | 1,mid+1,r);
        pushup(u);
    }
}

//u为根结点，查询l~r的区间和
int query(int u,int l,int r)
{
    //如果结点u的sum值都在查询范围内，则返回结点u的sum
    if(l<=tr[u].l&&r>=tr[u].r) return tr[u].sum;
    
    //如果结点u的sum值部分属于查询范围，那么递归查询这部分的sum值
    int res=0;
    int mid= tr[u].l + tr[u].r >>1;
    if(l<=mid) res+=query(u<<1,l,r);//错过：写成了query(u<<1,l,mid);
    if(r>mid) res+=query(u<<1 | 1,l,r);//注意l，r是查询范围，不能改变。
    
    return res;
}

//u为根结点，将第x个结点值加v
void modify(int u,int x,int v)
{
    //如果u为叶子结点，那么将该结点值加v
    if(tr[u].l==tr[u].r) tr[u].sum+=v;
    //否则，二分查找x结点
    else
    {
        int mid= tr[u].l + tr[u].r >> 1;
        if(x<=mid) modify(u<<1,x,v);
        else modify(u<<1 | 1,x,v);
        //更新完子节点后，更新u结点的sum值
        pushup(u);
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i) cin>>w[i];
    
    build(1,1,n);
    
    for(int i=1;i<=m;++i)
    {
        cin>>k>>a>>b;
        if(k==1) modify(1,a,b);
        else if(k==0) cout<<query(1,a,b)<<endl;
    }
    
    return 0;
}

例题

1.Acwing1265. 数星星

题目链接：1265. 数星星 - AcWing题库

注意点：将x的值都加1，不影响星星的等级。

因为星星x值可能为0，而树状数组是从下标为1开始的

#include<iostream>
using namespace std;

const int MAXN = 32010;

int n,a,b;
int tr[MAXN];
int cnt[MAXN];

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

void add(int x,int v)
{
    for(int i=x;i<=MAXN;i+=lowbit(i)) tr[i]+=v;
}

int query(int x)
{
    int res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}

int main()
{
   cin>>n;
   for(int i=0;i<n;++i)
   {
       cin>>a>>b;
       a++;
       cnt[query(a)]++;
       add(a,1);
   }
   
   for(int i=0;i<n;++i)
   {
       cout<<cnt[i]<<endl;
   }
   
   return 0;
}

2.Acwing 1270. 数列区间最大值

题目链接：1270. 数列区间最大值 - AcWing题库

注意点：query函数中要将res初始化为负无穷，因为未保证序列数都为正数

#include<iostream>
#include<cstdio>
#include<climits>
using namespace std;

const int MAXN = 100005;
int n,m;
int t,a,b;
int w[MAXN];

struct Node
{
    int l,r;
    int big;
}tr[MAXN*4];

void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[l]};
    else
    {
        tr[u]={l,r};
        int mid = r + l >>1;
        build(u<<1,l,mid);
        build(u<<1 | 1,mid+1,r);
        tr[u].big=max(tr[u<<1].big,tr[u<<1 | 1].big);//直接写，减少pushup函数调用时间
    }
}

int query(int u,int l,int r)
{
    if(l<=tr[u].l&&r>=tr[u].r) return tr[u].big;
    else
    {
        int res=INT_MIN;//初始化为负无穷
        int mid = tr[u].l + tr[u].r >> 1;
        if(l<=mid) res=query(u<<1,l,r);
        if(r>mid) res=max(res,query(u<<1 | 1,l,r));
        return res;
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i) cin>>w[i];
    
    build(1,1,n);
    
    for(int i=1;i<=m;++i)
    {
        scanf("%d%d",&a,&b);//不用cin，加快读取速度
        printf("%d\n",query(1,a,b));//不用cout，加快输出速度
    }
    
    return 0;
}