线段树模板
就去年暑假的时候老师讲了一点,然后当时也没怎么做题,现在又重新拿出来研究,先敲出来了一个区间和的模板放在这里,再慢慢深入。
#include<iostream> #include<cstdio> using namespace std; const int maxn = 1e5; int a[maxn],tree[maxn]; void build(int p,int l,int r)///p线段树上当前节点编号l当前结点区间的左界r右界 { if(l == r) { tree[p] = a[l]; return; } int mid = (l + r) / 2; build(p * 2,l,mid); build(p * 2 + 1,mid + 1,r); tree[p] = tree[p * 2] + tree[p * 2 + 1]; } void change(int p,int l,int r,int x,int num) { /*p当前节点编号 l当前节点区间的左界r是右界 x表示需要修改的节点编号 num表示对x增加的值*/ if(l == r) { tree[p] += num; return; } int mid = (l + r) / 2; if(x <= mid) change(p * 2,l,mid,x,num); else change(p * 2 + 1,mid + 1,r,x,num); tree[p] = tree[p * 2] + tree[p * 2 + 1]; } int find(int p,int l,int r,int x,int y) { /*在plr这棵子树里查询x到y的区间和*/ if(x <= l && r <= y) return tree[p]; int mid = (l + r) / 2; if(y <= mid) return find(p * 2,l,mid,x,y); if(x > mid) return find(p * 2 + 1,mid + 1,r,x,y); return find(p * 2,l,mid,x,mid) + find(p * 2 + 1,mid + 1,r,mid + 1,y); } int main() { int n; scanf("%d",&n); for(int i = 1;i <= n; i++) scanf("%d",&a[i]); build(1,1,n); //建树 int p,q; scanf("%d %d",&p,&q); //查询p到q的区间和 int sum = find(1,1,n,p,q); printf("%d\n",sum); int x,num; scanf("%d %d",&x,&num); //第x个节点加上num change(1,1,n,x,num); scanf("%d %d",&p,&q); sum = find(1,1,n,p,q); printf("%d\n",sum); return 0; }