【可持久化线段树】HDU 4348 To the moon

 HDU 4348 To the moon

• 题意：对一个数列，有三种操作。

1. 将区间[ql, qr]中的每个数都加上val。只有这个操作会改变数列的版本。
2. 查询当前版本[ql, qr]的元素和
3. 查询历史版本[ql, qr]的元素和
4. 返回某历史版本H，下一次操作从H开始。

• 思路：一看有各种版本，还是求区间和，那肯定是可持久化线段树。但是由于change操作，每修改一次元素值就要添加新的结点肯定会空间爆炸。所以我们选择新增一个变量add，用来记录当前区间的累加值。查询的时候以add作为参数，记录当前结点的祖先结点的累加值。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define INF 0x3f3f3f3f
#define lowbit(x) x & (-x)

#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
#define eps  1e-6

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxN = 1e5 + 5;
int n, m;
int arr[maxN];
struct node{
    int ls, rs;
    ll val, add;
    node() {}
    node(int a, int b, ll c, ll d) : ls(a), rs(b), val(c), add(d) {}
}tree[maxN * 30];
int root[maxN], tot, Version;
void pushup(int rt) { tree[rt].val = tree[tree[rt].ls].val + tree[tree[rt].rs].val; }
void build_tree(int &rt, int l, int r)
{
    rt = ++ tot;
    tree[rt] = node(0, 0, 0, 0);//初始化为空
    if(l == r) { tree[rt].val = arr[l]; return ; }
    int mid = MID;
    build_tree(tree[rt].ls, l, mid);
    build_tree(tree[rt].rs, mid + 1, r);
    pushup(rt);
}
void update(int &rt, int pre, int l, int r, int ql, int qr, ll val)
{
    rt = ++ tot;
    tree[rt] = tree[pre];
    tree[rt].val += val * (ll)(qr - ql + 1);//包括[ql, qr]的所有区间都加上add，避免用pushup。如果用pushup，就只有查询区间加了add，而大区间没有加
    if(ql == l && qr == r){ tree[rt].add += val; return ; }
    int mid = MID;
    if(qr <= mid)
        update(tree[rt].ls, tree[pre].ls, l, mid, ql, qr, val);
    else if(ql > mid)
        update(tree[rt].rs, tree[pre].rs, mid + 1, r, ql, qr, val);
    else
    {
        update(tree[rt].ls, tree[pre].ls, l, mid, ql, mid, val);//所以要保证[ql, qr]是有效区间
        update(tree[rt].rs, tree[pre].rs, mid + 1, r, mid + 1, qr, val);
    }
}
ll query(int rt, int l, int r, int ql, int qr, ll add)
{
    if(ql <= l && qr >= r)
        return tree[rt].val + add * (ll)(r - l + 1);
    add += tree[rt].add;//将祖先结点的add累加作为参数传递到其子孙区间
    int mid = MID;
    if(qr <= mid) return query(tree[rt].ls, l, mid, ql, qr, add);
    else if(ql > mid) return query(tree[rt].rs, mid + 1, r, ql, qr, add);
    else return query(tree[rt].ls, l, mid, ql, qr, add) + query(tree[rt].rs, mid + 1, r, ql, qr, add);
}
void back(int vers)
{
    if(root[vers + 1] && root[vers + 1] - tot != 1)//如果下一个版本存在,并且因为是多组输入，所以如果下一个版本根结点序号-tot != 1才能说明有下一个版本
        tot = root[vers + 1] - 1;
    Version = vers;
}
int main()
{
    while(~scanf("%d%d", &n, &m))
    {
        tot = 0; Version = 0;
        for(int i = 1; i <= n; i ++ )
            scanf("%d", &arr[i]);
        build_tree(root[0], 1, n);
        char op[2]; int l, r;
        while(m -- )
        {
            scanf("%s", op);
            if(op[0] == 'C')//change
            {
                scanf("%d%d", &l, &r);
                ll val; scanf("%lld", &val);
                ++ Version;
                update(root[Version], root[Version - 1], 1, n, l, r, val);
            }
            else if(op[0] == 'Q')//查询当前版本
            {
                scanf("%d%d", &l, &r);
                printf("%lld\n", query(root[Version], 1, n, l, r, 0));
            }
            else if(op[0] == 'H')//查询历史版本
            {
                scanf("%d%d", &l, &r);
                int vers; scanf("%d", &vers);
                printf("%lld\n", query(root[vers], 1, n, l, r, 0));
            }
            else if(op[0] == 'B')//返回历史版本
            {
                int v; scanf("%d", &v);
                back(v);
            }
        }
    }
   return 0;
}
/*
2 100
1 1
C 1 1 1
B 1
C 1 1 1
C 1 2 1
C 1 2 1
C 1 2 1
C 1 2 1
C 1 2 1
H 1 1 0
H 1 1 1
H 1 1 2
H 1 1 3
H 1 1 4
H 1 1 5
H 1 1 6
H 1 1 7
 */