HDU 4348 To the moon
- 将区间[ql, qr]中的每个数都加上val。只有这个操作会改变数列的版本。
- 查询当前版本[ql, qr]的元素和
- 查询历史版本[ql, qr]的元素和
- 返回某历史版本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-6using 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*/