HIHO线段树(成段)

#include <stdio.h>
#define lson l,mid,id<<1 
#define rson mid+1,r,id<<1|1
const int MM = 100001;
int num[MM<<2],lazy[MM<<2];

void push_down(int l,int r,int id)
{
    int mid=(l+r)>>1;
    num[id<<1]=lazy[id]*(mid-l+1);
    num[id<<1|1]=lazy[id]*(r-mid);
    lazy[id<<1]=lazy[id<<1|1]=lazy[id];
    lazy[id]=0;
}
void build_tree(int l,int r,int id)
{
    lazy[id]=0;
    if(l==r)
    {
        scanf("%d",&num[id]);
        return;
    }
    else
    {
        int mid=(l+r)>>1;
        build_tree(lson);
        build_tree(rson);
        num[id]=num[id<<1]+num[id<<1|1];
    }
}

void Update(int L,int R,int e,int l,int r,int id)
{
    if(L<=l&&r<=R)
    {
        num[id]=(r-l+1)*e;
        lazy[id]=e;
        return;
    }
    if(lazy[id])push_down(l,r,id);
    int mid=(l+r)>>1;
    if(L<=mid)
        Update(L,R,e,lson);
    if(R>mid)
        Update(L,R,e,rson);
    num[id]=num[id<<1]+num[id<<1|1];

}
int Query(int L,int R,int l,int r,int id)
{
    if(L<=l&&r<=R)
    {
        return num[id];
    }
    if(lazy[id])push_down(l,r,id);
    int mid=(l+r)>>1;
    int ret=0;
    if(L<=mid)ret+=Query(L,R,lson);
    if(R>mid)ret+=Query(L,R,rson);
    num[id]=num[id<<1]+num[id<<1|1];
    return ret;
}
int main()
{
    int n,m,op,x,y,z;
    scanf("%d",&n);
    build_tree(1,n,1);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d %d %d",&x,&y,&z);
            Update(x,y,z,1,n,1);
        }
        else 
        {
            scanf("%d %d",&x ,&y);
            int ans=Query(x,y,1,n,1);
            printf("%d
",ans );
        }
    }
}