该题之前用树状数组写过, 最近在学习线段树, 用线段树重新写了一遍。
在学习一种新的数据结构之前, 最重要的是要理解其结构是什么样子的, 这个可以参照《算法竞赛入门经典-训练指南》P199页。
比较重要的是理解好几个变量:
1. 每个结点有一个编号。 这个编号对应了其所统治的区间, 我们从线段树的祖先结点开始从上到下, 从左到右的顺序给所有结点编号,这样, 编号为i的结点其左右结点的编号就是2i和2i+1。
2.我们每次计算的m = (l + r) >> 1是当前区间[l, r]的中点, 为的是将当前区间不重复的分成两个子区间。 然后这两个子区间的结点编号又分别对应了2i 和 2i+1。
这样我们每次从祖先结点&&最大区间开始进行, 采用分治的思想, 就可以快速的进行计算了, 另外在更新操作的时候利用线段树的特点,每次递归结束之后顺带更新当前结点的值。
学习建议: 第一遍可以模仿, 建议模仿多个版本, 自己组建一个自己喜欢的代码风格,这样也有助于理解和记忆。
明白原理和过程之后, 以后尽量不要看模板, 自己手写出来, 这样可以加深理解。
细节参见代码:
#include<cstdio> #include<cstring> #include<algorithm> #include<iostream> #include<string> #include<vector> #include<stack> #include<bitset> #include<cstdlib> #include<cmath> #include<set> #include<list> #include<deque> #include<map> #include<queue> #define Max(a,b) ((a)>(b)?(a):(b)) #define Min(a,b) ((a)<(b)?(a):(b)) using namespace std; typedef long long ll; const double PI = acos(-1.0); const double eps = 1e-6; const int INF = 1000000000; const int maxn = 50000+10; int T,p,v,n,m,kase=0,sum[maxn<<2]; void push_up(int o) { sum[o] = sum[o<<1] + sum[o<<1|1]; } void build(int l, int r, int o) { int m = (l + r) >> 1; if(l == r) { scanf("%d",&sum[o]); return ; } build(l, m, o<<1); build(m+1, r, o<<1|1); push_up(o); } void update(int p, int add, int l, int r, int o) { int m = (l + r) >> 1; if(l == r) { sum[o] += add; return ; } if(p <= m) update(p, add, l, m, o<<1); else update(p, add, m+1, r, o<<1|1); push_up(o); } int query(int L, int R, int l, int r, int o) { int m = (l + r) >> 1, ans = 0; if(L <= l && r <= R) { return sum[o]; } if(m >= L) ans += query(L, R, l, m, o<<1); if(m+1 <= R) ans += query(L, R, m+1, r, o<<1|1); return ans; } int main() { scanf("%d",&T); while(T--) { scanf("%d",&n); build(1,n,1); char s[10]; printf("Case %d:\n",++kase); while(~scanf("%s",s)) { if(s[0] == 'E') break; scanf("%d%d",&p,&v); if(s[0] == 'A') update(p,v,1,n,1); else if(s[0] == 'S') update(p,-v,1,n,1); else printf("%d\n",query(p,v,1,n,1)); } } return 0; }