本文主要是介绍hdu1166线段树的点跟新模板,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define lson l, m, rt<<1//左子树的rt即当前节点在这个树中的位置(编号)为每次乘2
#define rson m+1, r, rt<<1|1//右子树树的当前节点的编号为乘2加1using namespace std;const int maxn = 55555;
int sum[maxn<<2];void pushup(int rt)
{sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}void build(int l, int r, int rt)//左右区间,以及当前所正在建造的节点
{if(l == r){scanf("%d", &sum[rt]);//表示当前节点值return ;}int m;m = (l+r)>>1;build(lson);//(define 的作用)递归建树build(rson);pushup(rt);//这里为从下向上的每个父亲节点的值,如若更新叶子节点则必通过这里向上更新父亲节点的值
}void update(int p, int add, int l, int r, int rt)
{if(l == r){sum[rt] += add;return ;}int m = (l+r)>>1;if(p <= m) update(p, add, lson);else update(p, add, rson);pushup(rt);
}int query(int newL, int newR, int l, int r, int rt)
{//cout << 1<< endl;if(newL <= l && newR >= r) return sum[rt];int m = (l+r)>>1;int ret = 0;if(newL <= m) ret += query(newL, newR, lson);//在左子树上所要求的左区间小于等于m就递归求和if(newR > m) ret += query(newL, newR, rson);// 在右子树上m大于所求区间的右区间就递归求和return ret;
}
int main()
{int T, n;//T多组输入,n总区间的大小scanf("%d", &T);for(int cas = 1; cas <= T; cas++){memset(sum, 0, sizeof(sum));cout << "Case " << cas << ":" << endl;scanf("%d", &n);//这里输入总区间大小后就直接建树,简述过程为输入一个值建一个节点所以在函数中第二次输入build(1, n, 1);//建树左右区间,以及每个节点while(1){string s;cin >> s;if(s == "End") break;int a, b;scanf("%d%d", &a, &b);if(s == "Query") cout << query(a, b, 1, n, 1) << endl;//给定区间求和else if(s == "Sub") update(a, -b, 1, n, 1);//单点减区间加一个负数else update(a, b, 1, n, 1);}}return 0;
}/*顺便再次膜拜一下HH神不然不会有这么简洁的代码,理解很清楚*/
这篇关于hdu1166线段树的点跟新模板的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!