[kuangbin带你飞]专题七 线段树 poj-3468(板子，区间和，查询区间和)

C - A Simple Problem with Integers POJ - 3468

AC

#include <iostream>
#include <cstdio>
#include <cstring>
#define For(i,x,y) for(register int i=(x); i<=(y); i++)
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int a[maxn];
int n,m;
char s[1000];
struct tree
{
    int l,r;
    ll pre,tag;
} t[maxn<<2];
void build(int p, int l, int r)
{
    t[p].l=l;
    t[p].r=r;
    if(l==r)
    {
        t[p].pre=a[l];
        return ;
    }
    int mid=t[p].l+(t[p].r-t[p].l)/2;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
    // cout<<t[p].pre<<' '<<p<<endl;
    t[p].tag=0;
}
inline void f(int p, int l, int r, ll add)
{
    t[p].pre+=add*(r-l+1);
    t[p].tag+=add;
}
void pushdown(int p)
{
    if(t[p].tag)
    {
        f(p<<1,t[p<<1].l,t[p<<1].r,t[p].tag);
        f(p<<1|1,t[p<<1|1].l,t[p<<1|1].r,t[p].tag);
        t[p].tag=0;
    }
}
void update(int p, int x,int y, int z)
{
    if(x<=t[p].l && y>=t[p].r)
    {
        t[p].pre+=(ll)z*(t[p].r-t[p].l+1);
        t[p].tag+=z;
        return ;
    }
    pushdown(p);
    int mid=t[p].l+(t[p].r-t[p].l)/2;
    if(x<=mid)update(p<<1,x,y,z);
    if(y>mid)update(p<<1|1,x,y,z);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
ll query(int p, int x, int y)
{
    if(x<=t[p].l && y>=t[p].r)return t[p].pre;
    pushdown(p);
    int mid=t[p].l+(t[p].r-t[p].l)/2;
    ll ans=0;
    if(x<=mid)ans+=query(p<<1,x,y);
    // cout<<ans<<endl;
    if(y>mid)ans+=query(p<<1|1,x,y);
    //cout<<ans<<endl;
    return ans;
}
int main()
{
    scanf("%d%d", &n, &m);
    For(i,1,n)scanf("%d", &a[i]);
    build(1,1,n);
    For(i,1,m)
    {
        int x,y;
        ll add;
        scanf("%s", s);
        scanf("%d%d",&x,&y);
        if(s[0]=='C')
        {
            scanf("%d", &add);
            update(1,x,y,add);
        }
        else if(s[0]=='Q')printf("%lld\n",query(1,x,y));
    }
    return 0;
}