HDU 4348 To the moon（主席树区间修(加)）

To the moon(戳我看题)

题意：给你一串序列，每次对它有四种操作， 1. 将l到r之间所有的数+d，同时时间戳+1。 2. 查询当前时间戳下l到r的所有数的总和。 3.查询时间戳d下的l到r的所有数的和。 4.将时间戳返回至t。

思路：涉及历史查询，一眼主席树的题目，关键的如何进行区间修改。

主席树与线段树不同之处就是  每一棵树是不能修改的，也就是不能够pushDown。那么能否一直新建节点？答案是不可以，如果用新建节点来替代pushDown操作，那么每一次都会一直新建到叶子节点，空间肯定是撑不住的。

主席树区间修一个很巧妙的方法：更新的时候和普通线段树一样，pushup 时由于没有pushDown 子树是不会更新的，所以用自己的 lazy 更新就可以了.也就是 sum[i] = sum[lson[i]] + sum[rson[i]] + lazy[i] * (r - l + 1)。查询的时候，将每一个节点的 lazy 值放在函数参数上带着，解决了pushDown的问题。

Code：

#include<bits/stdc++.h>
#define debug(x) cout << "[" << #x <<": " << (x) <<"]"<< endl
#define pii pair<int,int>
#define clr(a,b) memset((a),b,sizeof(a))
#define rep(i,a,b) for(int i = a;i < b;i ++)
#define pb push_back
#define MP make_pair
#define LL long long
#define ull unsigned LL
#define ls i << 1
#define rs (i << 1) + 1
#define fi first
#define se second
#define ptch putchar
#define CLR(a) while(!(a).empty()) a.pop()

using namespace std;
inline LL read() {
    LL s = 0,w = 1;
    char ch = getchar();
    while(!isdigit(ch)) {
        if(ch == '-') w = -1;
        ch = getchar();
    }
    while(isdigit(ch))
        s = s * 10 + ch - '0',ch = getchar();
    return s * w;
}
inline void write(LL x) {
    if(x < 0)
        putchar('-'), x = -x;
    if(x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
}

const int maxn = 1e5 + 10;
const int M = maxn * 40;
LL C[M],lazy[M];
int lson[M],rson[M];
int tot,T[maxn];

void build(int &i,int l,int r){
    i = ++ tot;
    C[i] = lazy[i] = 0;
    if(l == r){
        LL x = read();
        C[i] = x;
        return ;
    }
    int mid = (l + r) >> 1;
    build(lson[i],l,mid);
    build(rson[i],mid + 1,r);
    C[i] = C[lson[i]] + C[rson[i]];
}

void update(int l,int r,int ul,int ur,int &i,int last,LL val){
    C[++ tot] = C[last];
    lson[tot] = lson[last];
    rson[tot] = rson[last];
    lazy[tot] = lazy[last];
    i = tot;
    if(ul <= l && r <= ur){
        C[i] += (r - l + 1) * val;
        lazy[i] += val;
        return ;
    }
    int mid = (l + r) >> 1;
    if(ul <= mid) update(l,mid,ul,ur,lson[i],lson[last],val);
    if(ur > mid) update(mid + 1,r,ul,ur,rson[i],rson[last],val);
    C[i] = C[lson[i]] + C[rson[i]] + (r - l + 1) * lazy[i];
}

LL query(int l,int r,int ql,int qr,int i,LL add){
    if(ql <= l && r <= qr){
        return C[i] + add * (r - l + 1);
    }
    int mid = (l + r) >> 1;
    LL ans = 0;
    if(ql <= mid) ans += query(l,mid,ql,qr,lson[i],add + lazy[i]);
    if(qr > mid) ans += query(mid + 1,r,ql,qr,rson[i],add + lazy[i]);
    return ans;
}

int main() {
    int n = read(),m = read();
    build(T[0],1,n);
    int now = 0;
    while(m --){
        char op; scanf(" %c",&op);
        if(op == 'C'){
            int l = read(),r = read(),d = read();
            update(1,n,l,r,T[now + 1],T[now],d);
            ++ now;
        }
        else if(op == 'Q'){
            int l = read(),r = read();
            write(query(1,n,l,r,T[now],lazy[now])); ptch('\n');
        }
        else if(op == 'H'){
            int l = read(),r = read(),t = read();
            write(query(1,n,l,r,T[t],lazy[t])); ptch('\n');
        }
        else {
            int t = read();
            now = t;
        }
    }
    return 0;
}