題目連結

題目描述：

如題，已知一個數列，你需要進行下面兩種操作：

1.將某區間每一個數加上x

2.求出某區間每一個數的和

輸入格式：

第一行包含兩個整數N、M，分別表示該數列數字的個數和操作的總個數。

第二行包含N個用空格分隔的整數，其中第i個數字表示數列第i項的初始值。

接下來M行每行包含3或4個整數，表示一個操作，具體如下：

操作1： 格式：1 x y k 含義：將區間[x,y]內每個數加上k

操作2： 格式：2 x y 含義：輸出區間[x,y]內每個數的和

輸出格式：

輸出包含若干行整數，即為所有操作2的結果。

線段樹模板，沒什麼好說的

建樹:

struct Node
{
    int
 
 l,r,len;
    ll val;
    int lazy;//sign
}tree[N*2];
void Maketree(int o,int l,int r)
{
    tree[o].l=l,tree[o].r=r,tree[o].len=r-l+1;
    if(l==r) tree[o].val=data[l];
    else
    {
        int mid=(l+r)>>1,nxt=o<<1;
        Maketree(nxt,l,mid);
        Maketree(nxt+1,mid+1,r);
        tree[
 
o].val=tree[nxt].val+tree[nxt+1].val;
    }
}

更新資料:

void pushdown(int o)//將lazy標記下傳到左右子樹
{
    if(!tree[o].lazy) return ;
    int p=tree[o].lazy,nxt=o<<1;
    tree[nxt].lazy+=p;
    tree[nxt+1].lazy+=p;
    tree[nxt].val+=p*tree[nxt].len;
    tree[nxt+1].val+=p*tree[nxt+1].len;
    tree[o].lazy=
 
0;
}

在[l,r]加上addval:

void update(int o,int l,int r,int addval)
{
    if(l<=tree[o].l && r>=tree[o].r)
    {
        tree[o].lazy+=addval;
        tree[o].val+=tree[o].len*addval;
        return ;
    }
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(l<=mid) update(nxt,l,min(r,mid),addval);
    if(r>mid) update(nxt+1,max(l,mid+1),r,addval);
    tree[o].val=tree[nxt].val+tree[nxt+1].val;
}

查詢：

ll query(int o,int l,int r)
{
    if(tree[o].l==l && tree[o].r==r) return tree[o].val;
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(r<=mid) return query(nxt,l,r);
    else if(l>mid) return query(nxt+1,l,r);
    return query(nxt,l,mid)+query(nxt+1,mid+1,r);
}

完整程式（醜到不行）：

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int n,m;
const int SIZE=1<<15;
inline char getch(){
    static char buf[SIZE],*p1=buf,*p2=buf;
    return p1==p2 && (p2=(p1=buf)+fread(buf,1,SIZE,stdin),p1==p2) ? EOF : *p1++;
}
int read(){
    int s=0,f=1;
    register char c=getch();
    while(c>'9'|| c<'0') { if(c=='-') f=-1;c=getch();}
    while(c<='9' && c>='0') {s=(s<<3)+(s<<1)+c-'0';c=getch();}
    return s*=f;
}

struct Node
{
    int l,r,len;
    ll val;
    int lazy;//sign
}tree[N*2];
int data[N];
void Maketree(int o,int l,int r)
{
    tree[o].l=l,tree[o].r=r,tree[o].len=r-l+1;
    if(l==r) tree[o].val=data[l];
    else
    {
        int mid=(l+r)>>1,nxt=o<<1;
        Maketree(nxt,l,mid);
        Maketree(nxt+1,mid+1,r);
        tree[o].val=tree[nxt].val+tree[nxt+1].val;
    }
}

void pushdown(int o)
{
    if(!tree[o].lazy) return ;
    int p=tree[o].lazy,nxt=o<<1;
    tree[nxt].lazy+=p;
    tree[nxt+1].lazy+=p;
    tree[nxt].val+=p*tree[nxt].len;
    tree[nxt+1].val+=p*tree[nxt+1].len;
    tree[o].lazy=0;
}

void update(int o,int l,int r,int addval)
{
    if(l<=tree[o].l && r>=tree[o].r)
    {
        tree[o].lazy+=addval;
        tree[o].val+=tree[o].len*addval;
        return ;
    }
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(l<=mid) update(nxt,l,min(r,mid),addval);
    if(r>mid) update(nxt+1,max(l,mid+1),r,addval);
    tree[o].val=tree[nxt].val+tree[nxt+1].val;
}
ll query(int o,int l,int r)
{
    if(tree[o].l==l && tree[o].r==r) return tree[o].val;
    int mid=(tree[o].l+tree[o].r)>>1,nxt=o<<1;
    pushdown(o);
    if(r<=mid) return query(nxt,l,r);
    else if(l>mid) return query(nxt+1,l,r);
    return query(nxt,l,mid)+query(nxt+1,mid+1,r);
}

int main()
{
    int opt,x,y,k;
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        data[i]=read();
    Maketree(1,1,n);
    while(m--)
    {
        opt=read();
        if(opt==1)
        {
            x=read(),y=read(),k=read();
            update(1,x,y,k);
        }
        if(opt==2)
        {
            x=read(),y=read();
            printf("%lld\n",query(1,x,y));
        }
//		for(int i=1;tree[i].val!=0;i++)
//		    printf("tree[%d].val=%d ",i,tree[i].val);
//		printf("\n");
    }
    return 0;
}