线段树加乘操作 求区间和和平方和

搞了整整一晚上加一下午 应该是懂了

加乘区间和别人的题解清晰易懂:这里

 

平方和

题目链接

 

在线段树中除了标记和区间和的值我们再保存一个区间平方和的值 然后推得对平方和进行修改的表达式

 

注意：

1.复制上一行粘贴到下一行的时候一定要注意是不是所有东西都改了 lson和rson的长度不一样 找了很久

2.传参的时候要清楚要传的到底是什么pushdown(1,l,r)找了很久

3.query的时候也要pushdown 因为只有这样才能让儿子进行完整的计算

 

#include 
#include 
#define ll long long
using namespace std;
const int N=100100;
ll a[N];
struct node{
    ll mul,add,val,val2;
}segTree[N<<2];
inline int lson(int x){return x<<1;}
inline int rson(int x){return lson(x)|1;}
void build(int t,int l,int r)
{
    segTree[t].add=0;
    segTree[t].mul=1;
    if(l==r)
    {
        segTree[t].val=a[l];
        segTree[t].val2=a[l]*a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson(t),l,mid);
    build(rson(t),mid+1,r);
    segTree[t].val=segTree[lson(t)].val+segTree[rson(t)].val;
    segTree[t].val2=segTree[lson(t)].val2+segTree[rson(t)].val2;
}
void pushdown(int t,int l,int r)
{
    int mid=(l+r)>>1;
    segTree[lson(t)].add=segTree[lson(t)].add*segTree[t].mul+segTree[t].add;
    segTree[rson(t)].add=segTree[rson(t)].add*segTree[t].mul+segTree[t].add;
    segTree[lson(t)].mul=segTree[lson(t)].mul*segTree[t].mul;
    segTree[rson(t)].mul=segTree[rson(t)].mul*segTree[t].mul;

    segTree[lson(t)].val2=segTree[lson(t)].val2*segTree[t].mul*segTree[t].mul
    +(segTree[t].add*segTree[t].add)*(mid-l+1)+2*segTree[t].add*segTree[t].mul*segTree[lson(t)].val;

    segTree[rson(t)].val2=segTree[rson(t)].val2*segTree[t].mul*segTree[t].mul
    +(segTree[t].add*segTree[t].add)*(r-mid)+2*segTree[t].add*segTree[t].mul*segTree[rson(t)].val;

    segTree[lson(t)].val=segTree[lson(t)].val*segTree[t].mul+segTree[t].add*(mid-l+1);
    segTree[rson(t)].val=segTree[rson(t)].val*segTree[t].mul+segTree[t].add*(r-mid);


    segTree[t].mul=1;
    segTree[t].add=0;

}
///*
void update1(int t,int l,int r,int s,int e,ll c)
{
    if(l>e||r=s&&r<=e)///完全在里面
    {
        segTree[t].mul=segTree[t].mul*c;
        segTree[t].val=segTree[t].val*c;
        segTree[t].val2=segTree[t].val2*c*c;
        segTree[t].add=segTree[t].add*c;
        return;
    }
    pushdown(t,l,r);
    int mid=(l+r)>>1;
    update1(lson(t),l,mid,s,e,c);
    update1(rson(t),mid+1,r,s,e,c);
    segTree[t].val=segTree[lson(t)].val+segTree[rson(t)].val;
    segTree[t].val2=segTree[lson(t)].val2+segTree[rson(t)].val2;
}
///+
void update2(int t,int l,int r,int s,int e,ll c)
{
    if(l>e||r=s&&r<=e)///完全在里面
    {
        segTree[t].val2=segTree[t].val2+2*c*segTree[t].val+c*c*(r-l+1);
        segTree[t].val=segTree[t].val+c*(r-l+1);
        segTree[t].add=segTree[t].add+c;
        return;
    }
    pushdown(t,l,r);
    int mid=(l+r)>>1;
    update2(lson(t),l,mid,s,e,c);
    update2(rson(t),mid+1,r,s,e,c);
    segTree[t].val=segTree[lson(t)].val+segTree[rson(t)].val;
    segTree[t].val2=segTree[lson(t)].val2+segTree[rson(t)].val2;
}
ll query1(int t,int l,int r,int s,int e)
{
    if(l>e||r=s&&r<=e)///完全在里面
    {
        return segTree[t].val;
    }
    pushdown(t,l,r);
    int mid=(l+r)>>1;
    return query1(lson(t),l,mid,s,e)+query1(rson(t),mid+1,r,s,e);
}
ll query2(int t,int l,int r,int s,int e)
{
    if(l>e||r=s&&r<=e)///完全在里面
    {
        return segTree[t].val2;
    }
    pushdown(t,l,r);
    int mid=(l+r)>>1;
    return query2(lson(t),l,mid,s,e)+query2(rson(t),mid+1,r,s,e);
}
int main()
{
    int n,op,m,x,y;
    ll c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
        }
        build(1,1,n);
        while(m--)
        {
            scanf("%d",&op);
            if(op==3)
            {
                scanf("%d%d%lld",&x,&y,&c);
                update1(1,1,n,x,y,c);
            }
            else if(op==4)
            {
                scanf("%d%d%lld",&x,&y,&c);
                update2(1,1,n,x,y,c);
            }
            else if(op==1)
            {
                scanf("%d%d",&x,&y);
                printf("%lld\n",query1(1,1,n,x,y));
            }
            else if(op==2)
            {
                scanf("%d%d",&x,&y);
                printf("%lld\n",query2(1,1,n,x,y));
            }
        }
    }
    return 0;
}