243. 一个简单的整数问题2树状数组 区间查询 区间修改

Posted 2021-10-30 幽殇默

https://www.acwing.com/problem/content/244/

上图摘自acwing的铅笔佬

#include<bits/stdc++.h>
using namespace std;
typedef long long int LL;
const int N=1e5+10;
LL tr1[N],tr2[N],n,m,a[N];
int lowbit(int x){return x&(-x);}
LL add(LL tr[],LL x,LL c)
{
    for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}
LL query(LL tr[],LL x)
{
    LL res=0;
    for(int i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}
LL sum(int x)
{
    return query(tr1,x)*(x+1)-query(tr2,x);//不全的正方型的减去 补上的
}
int main(void)
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        LL b=a[i]-a[i-1];
        add(tr1,i,b);
        add(tr2,i,i*b);
    }
    while(m--)
    {
        string op; cin>>op;
        if(op=="C")
        {
            int l,r,d; cin>>l>>r>>d;
            add(tr1,l,d),add(tr1,r+1,-d);
            add(tr2,l,l*d),add(tr2,r+1,(r+1)*-d);
        }
        else 
        {
            int l,r; cin>>l>>r;
            cout<<sum(r)-sum(l-1)<<endl;
        }
    }
    return 0;
}