luogu P3372 【模板】线段树 1-CSDN博客

本文深入探讨了树状数组的高级应用技巧，通过两层树状数组实现区间查询与区间修改，有效处理动态数组更新及求和问题。文章详细介绍了如何通过差分维护实线区间修改，并给出具体算法实现，包括关键的加法交换律与结合律推导过程，以及查询和修改操作的代码示例。

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

树状数组套树状数组为树套树做铺垫

/*
    树状数组套树状数组 区间查询与区间修改
    差分一下 维护 实线区间修改
    a[1]+a[2]+……+a[r-1]+a[r]
    用上方公式推导得出
    =s[1]+(s[1]+s[2])+……+(s[1]+……+s[r])
    根据加法交换律与结合律：
    =(s[1]*(r))+(s[2]*(r-1))+……(s[r]*1)
    那么：
    =r*(s[1]+s[2]+……+s[r])
        -(s[1]*0+s[2]*1+……+s[r]*(r-1))
    再来一个树状数组 S[i] 维护 s[i]*(i-1)
    查询就是 r*find_s(r) - find_S(r)
*/
#include<cstdio>
#define lowbit(x) (x&-x)
using namespace std; typedef long long ll;
const int N=200050; ll s[N],S[N],res; int n,m,a[N];
void add(ll*s,int x,ll c) {while(x<=n) s[x]+=c,x+=lowbit(x);} 
ll find(ll*s,int x) {res=0;while(x) res+=s[x],x-=lowbit(x);return res;} 
int main()
{
    scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]),
        add(s,i,a[i]-a[i-1]),add(S,i,1LL*(a[i]-a[i-1])*(i-1));
    int op,x,y,k; scanf("%d",&m);while(m--)
    {
        scanf("%d",&op); if(op==1)
        {
            scanf("%d%d%d",&x,&y,&k);add(s,x,k),add(s,y+1,-k);
            add(S,x,k*(x-1)),add(S,y+1,-k*y);continue;
        }
        scanf("%d%d",&x,&y);
        printf("%lld\n",y*find(s,y)+find(S,x-1)-(x-1)*find(s,x-1)-find(S,y));
    }
    return 0;
}