【线段树】POJ 3468 A Simple Problem with Integers

POJ 3468 A Simple Problem with Integers

• 题意：区间修改，区间查询
• 思路：模板题

！！！！！！！！！！！！！！这道题忘记了在query里边pushdown懒标记，也没有开long long，不仅debug了很久，还wa了一发

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>
#define INF 0x3f3f3f3f
#define P(x) x>0 ? x : 0
#define MID l + r >> 1
#define lsn rt << 1
#define rsn rt << 1 | 1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr

using namespace std;
typedef long long ll;
typedef vector<int>:: iterator VITer;
const int maxN=1e5+5;
ll N,Q;
ll num[maxN];
ll tree[maxN<<2],lazy[maxN<<2];

void pushup(ll rt) { tree[rt] = tree[lsn]+tree[rsn]; }

void build_tree(ll rt,ll l,ll r)
{
    if(l==r) { tree[rt] = num[l]; return; }
    ll mid=MID;
    build_tree(Lson);
    build_tree(Rson);
    pushup(rt);
}

void pushdown(ll rt,ll l,ll r)
{
    if(lazy[rt])
    {
        ll mid=MID;
        tree[lsn] += lazy[rt]*(mid-l+1);
        tree[rsn] += lazy[rt]*(r-mid);
        lazy[lsn] += lazy[rt];
        lazy[rsn] += lazy[rt];
        lazy[rt]=0;
    }
}

void update_range(ll rt,ll l,ll r,ll ql,ll qr,ll val)
{
    if(ql<=l && qr>=r) { tree[rt]+=val*(r-l+1); lazy[rt]+=val; return; }
    pushdown(rt,l,r);
    ll mid=MID;
    if(qr<=mid) update_range(QL,val);
    else if(ql>mid) update_range(QR,val);
    else { update_range(QL,val); update_range(QR,val); }
    pushup(rt);
}

ll query(ll rt,ll l,ll r,ll ql,ll qr)
{
    if(ql <= l && qr >= r) { return tree[rt]; }
    pushdown(rt,l,r);
    ll mid=MID;
    if(qr <= mid) return query(QL);
    else if(ql > mid) return query(QR);
    else return query(QL)+query(QR);
}

void init()
{
    memset(lazy, 0, sizeof(lazy));
}

int main()
{
    while(~scanf("%lld%lld", &N, &Q))
    {
        for(ll i=1; i<=N ; i++)
            scanf("%lld", &num[i]);
        init();
        build_tree(1,1,N);
        while(Q--)
        {
            char s[10];
            scanf("%s",s);
            if(s[0]=='C')
            {
                ll a,b,c;
                scanf("%lld%lld%lld",&a, &b, &c);
                update_range(1,1,N,a,b,c);
            }
            else if(s[0]=='Q')
            {
                ll aa,bb;
                scanf("%lld%lld",&aa,&bb);
                printf("%lld\n",query(1,1,N,aa,bb));
            }
        }
    }
    return 0;
}