标签:
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 59891 Accepted Submission(s): 25331
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <ctype.h> #include <math.h> #include <queue> #include <iostream> #include <algorithm> #define INF 0x3f3f3f3f using namespace std; struct node { int ll, rr; int mm;//保存区间和 }q[200000+10]; int fa[50000+2]; /* 线段树是一种树状的数据结构,但在算法的实现过程中, 用的是一维结构体数组来模拟的, */ void Build(int i, int ll, int rr) { q[i].ll=ll; q[i].rr=rr;//当前线段节点的左右区间 q[i].mm=0; if(ll==rr){ fa[ll]=i; return; }//如果当前的是叶子节点 //fa[]记录下了每个叶子节点的存储位置 Build(i*2, ll, (ll+rr)/2); Build(i*2+1, (ll+rr)/2+1, rr); } //往下递归两个儿子节点 void update(int ri) { if(ri==1){ return ; }//到达根节点 修改完成返回 int fi=ri/2; q[fi].mm = q[fi*2].mm + q[fi*2+1].mm; update(ri/2); } int ans; void query(int i, int ll ,int rr) { if(q[i].ll==ll&&q[i].rr==rr){ ans=ans+q[i].mm; //printf("%d--\n", ans ); return; }// i=i*2; if(ll<=q[i].rr){//在这个子节点有范围 if(rr<=q[i].rr){ query(i, ll, rr); }else{ query(i, ll, q[i].rr); } } i=i+1; if(rr>=q[i].ll){ if(ll>=q[i].ll ){ query(i, ll, rr); }else{ query(i, q[i].ll, rr); } } } int main() { int tg; scanf("%d", &tg); int cnt=1; int n, u, v; char s[20]; while(tg--){ scanf("%d", &n); Build(1, 1, n);//建树 int cur; for(int i=1; i<=n; i++){ scanf("%d", &cur); q[fa[i]].mm=cur; update(fa[i]); } printf("Case %d:\n", cnt++); while(scanf("%s", s)!=EOF) { if(s[0]==‘E‘) break; else if(s[0]==‘Q‘){ scanf("%d %d", &u, &v); /* for(int i=1; i<=10; i++) printf("%d ", q[fa[i]].mm ); printf("\n\n"); */ ans=0; query(1, u, v); printf("%d\n", ans); } else if(s[0]==‘A‘){ scanf("%d %d", &u, &v); q[fa[u]].mm+=v; update(fa[u]); } else if(s[0]==‘S‘){ scanf("%d %d", &u, &v); q[fa[u]].mm-=v; update(fa[u]); } } } return 0; }
HDU 1166 敌兵布阵 【线段树-点修改--计算区间和】
标签:
原文地址:http://www.cnblogs.com/yspworld/p/4738709.html