当前位置 : 主页 > 网页制作 > html >

线段树简单操作模板复习(本来就不熟练)

来源:互联网 收集:自由互联 发布时间:2021-06-12
// 参考博客https://www.cnblogs.com/TheRoadToTheGold/p/6254255.html#4175712 // 懒人标记: 表示当前结点区间值已经改变,但是下面的区间还没改变。每次询问到有懒人标记的结点时,在进一步询问他

// 参考博客 https://www.cnblogs.com/TheRoadToTheGold/p/6254255.html#4175712

// 懒人标记: 表示当前结点区间值已经改变,但是下面的区间还没改变。每次询问到有懒人标记的结点时,在进一步询问他的子节点时,下放标记(下放的同时改变了子结点的值,并且赋予子节点懒人标记)
// 因此 当将要查询对应区间时候 之前修改区间的操作在此时才进行 -- 因此叫懒人标记
1
#include<cstdio> 2 using namespace std; 3 4 const int MAXN = 100000; 5 6 int ans;// 每次查询的结果 7 struct node { 8 int l, r, w, f;// 左端点下标,右端点下标,l~r各点权值和,"懒人"标记(在修改区间值时候,‘手动‘修改懒人标记) 9 }tree[4*MAXN+1]; 10 11 inline void build(int k, int ll, int rr) {// 建树 12 tree[k].l = ll, tree[k].r = rr; 13 if(tree[k].l == tree[k].r) { 14 scanf("%d", &tree[k].w); 15 return ; 16 } 17 int m = (ll + rr) / 2; 18 build(k*2, ll, m); 19 build(k*2+1, m+1, rr); 20 tree[k].w = tree[k*2].w + tree[k*2+1].w; 21 } 22 23 inline void down(int k) {// "懒"标记下传 24 tree[k*2].f += tree[k].f; 25 tree[k*2+1].f += tree[k].f; 26 tree[k*2].w += tree[k].f * (tree[k*2].r - tree[k*2].l + 1); 27 tree[k*2+1].w += tree[k].f * (tree[k*2+1].r - tree[k*2+1].l + 1); 28 tree[k].f = 0;// 表明下传过了 不能重复下传(下次查询到此不再下传) 29 } 30 31 inline void query_point(int k, int x) {// 单点查询(第x位置) 32 if(tree[k].l == tree[k].r) { 33 ans = tree[k].w; 34 return ; 35 } 36 if(tree[k].f) down(k); 37 int m = (tree[k].l + tree[k].r) / 2; 38 if(x <= m) query_point(k*2, x); 39 else query_point(k*2+1, x); 40 } 41 42 inline void change_point(int k, int x, int v) {// 单点修改 x->x+v 43 if(tree[k].l == tree[k].r) { 44 tree[k].w += v; 45 return ; 46 } 47 if(tree[k].f) down(k); 48 int m = (tree[k].l + tree[k].r) / 2; 49 if(x <= m) change_point(k*2, x, v); 50 else change_point(k*2+1, x, v); 51 tree[k].w = tree[k*2].w + tree[k*2+1].w; 52 } 53 54 inline void query_interval(int k, int x, int y) {// 区间查询 x~y之和 55 if(tree[k].l >= x && tree[k].r <= y) { 56 ans += tree[k].w; 57 return ; 58 } 59 if(tree[k].f) down(k); 60 int m = (tree[k].l + tree[k].r) / 2; 61 if(x <= m) query_interval(k*2, x, y); 62 if(y > m) query_interval(k*2+1, x, y); 63 } 64 65 inline void change_interval(int k, int x, int y, int v) {// 区间修改 x~y每个点+v 66 if(tree[k].l >= x && tree[k].r <= y) { 67 tree[k].w += (tree[k].r - tree[k].l + 1) * v;// 改变区间值 68 tree[k].f += v;// 懒人标记 69 return ; 70 } 71 if(tree[k].f) down(k); 72 int m = (tree[k].l + tree[k].r) / 2; 73 if(x <= m) change_interval(k*2, x, y, v); 74 if(y > m) change_interval(k*2+1, x, y, v); 75 tree[k].w += tree[k*2].w + tree[k*2+1].w; 76 } 77 78 int main() { 79 int n, m; 80 scanf("%d", &n); 81 build(1, 1, n); 82 scanf("%d", &m); 83 int p; 84 int x, y, v; 85 for(int i = 0; i != m; ++i) { 86 scanf("%d", &p); 87 ans = 0; 88 if(p == 1) { 89 scanf("%d", &x); 90 query_point(1, x);// 单点查询 输出第k个数 91 printf("%d\n", ans); 92 } 93 else if(p == 2) { 94 scanf("%d%d", &x, &v); 95 change_point(1, x, v);// 单点修改 x->x+v 96 } 97 else if(p == 3) { 98 scanf("%d%d", &x, &y); 99 query_interval(1, x, y);// 区间查询 x~y之和 100 printf("%d\n", ans); 101 } 102 else { 103 scanf("%d%d%d", &x, &y, &v); 104 change_interval(1, x, y, v);// 区间修改 x~y每个点+v 105 } 106 } 107 return 0; 108 }
网友评论