标签:
第一个线段树,参考别人代码
#include<iostream> #include<cctype> #include<cstdlib> #include<cstring> #include<cstdio> using 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 mid = (l + r) >> 1; build(l, mid, rt << 1); build(mid + 1, r, rt << 1 | 1); PushUp(rt); } void update(int p, int add, int l, int r, int rt) { if(l == r) { sum[rt] += add; return; } int mid = (l + r) >> 1; if(p <= mid) update(p, add, l, mid, rt << 1); else update(p, add, mid + 1, r, rt << 1 | 1); PushUp(rt); } int query(int ll, int rr, int l, int r, int rt) { if(ll <= l && rr >= r) return sum[rt]; int mid = (l + r) >> 1; int ret = 0; if(ll <= mid) ret += query(ll, rr, l, mid, rt << 1); if(rr > mid) ret += query(ll, rr, mid +1, r, rt << 1 | 1); return ret; } int main() { int t,c; char d[10]; scanf("%d", &t); for(c = 1; c <= t; c++){ printf("Case %d:\n", c); int n; scanf("%d", &n); build(1,n,1); while(scanf("%s",d) != EOF) { if(d[0] == ‘E‘) break; int x,y; scanf("%d%d",&x,&y); if(d[0] == ‘Q‘) { int ans = query(x, y, 1, n, 1); printf("%d\n",ans); } if(d[0] == ‘S‘) update(x,-y,1,n,1); if(d[0] == ‘A‘) update(x, y,1,n,1); } } return 0; }
#include<iostream> #include<cctype> #include<cstdlib> #include<cstring> #include<cstdio> using namespace std; struct seg { int l; int r; int n; }T[150011]; void build(int l, int r, int k) { int mid; if(l == r){ T[k].l = l; T[k].r = r; T[k].n = 0; return ; } mid = (l + r)/2; T[k].l = l; T[k].r = r; T[k].n = 0; build(l,mid,2*k); build(mid+1,r,2*k+1); } void insert(int n,int d,int k) { int mid; if(T[k].l == T[k].r && T[k].l == d) { T[k].n += n; return; } mid = (T[k].l + T[k].r)>>1; if(d <= mid) insert(n,d,2*k); else insert(n,d,2*k + 1); T[k].n = T[2*k].n + T[2*k + 1].n; } int ans; void search(int l, int r, int k) { int mid; if(T[k].l == l && T[k].r == r) { ans += T[k].n; return; } mid = (T[k].l + T[k].r) >> 1; if(r <= mid) search(l,r,2*k); else if(l > mid) search(l,r,2*k +1); else { search(l,mid,2*k); search(mid + 1,r,2*k+1); } } int main(){ int Case,TT; int n; int i; int temp; char str[11]; int a,b; scanf("%d",&TT); for(Case = 1;Case <= TT; Case++) { scanf("%d",&n); build(1,n,1); for(i = 1; i <= n; i++) { scanf("%d",&temp); insert(temp,i,1); } printf("Case %d:\n",Case); while(scanf("%s",str),strcmp(str,"End")) { scanf("%d%d",&a,&b); if(strcmp(str,"Add")==0) insert(b,a,1); else if(strcmp(str,"Sub")==0) insert(-b,a,1); else{ ans = 0; search(a,b,1); printf("%d\n",ans); } } } return 0; }
标签:
原文地址:http://www.cnblogs.com/yong-hua/p/4655487.html