hdu 1166 敌兵布阵(线段树基础题)
学习线段树~~~~~~~~~~~~要好好理解
此题是单点更新的线段树,考虑基本的询问,更新。
#include <iostream> #include <algorithm> #include <cmath> #include <cstdio> #include <cstdlib> #include <cstring> #include <string> #include <vector> #include <set> #include <queue> #include <stack> #include <climits>//形如INT_MAX一类的 #define MAX 50005 #define INF 0x7FFFFFFF #define REP(i,s,t) for(int i=(s);i<=(t);++i) #define ll long long #define mem(a,b) memset(a,b,sizeof(a)) #define mp(a,b) make_pair(a,b) # define eps 1e-5 //#pragma comment(linker, "/STACK:36777216") ///传说中的外挂 using namespace std; struct node { int left,right,mid,value,cover; } edge[4*MAX]; void build(int l,int r,int num) //l,r为当前节点的左右端点,num为当前节点在数组中的下标 { edge[num].left = l; edge[num].right = r; edge[num].value = 0; edge[num].mid = (l+r) >> 1; if(l != r) // 如果不是叶子节点 { build(l,edge[num].mid,num * 2); //左子树 build(edge[num].mid + 1,r,num*2+1); //右子树 } } void update(int num,int mid,int v) //单点更新 mid为更新点,v为更新值 { if(edge[num].left == mid && edge[num].right == mid) { edge[num].value += v; return ; } if(edge[num].mid >= mid) update(2*num,mid,v); else update(2*num+1,mid,v); edge[num].value = edge[num*2].value + edge[num*2+1].value; // 注意是‘=’不是‘+=’,更新值 } int query(int l,int r,int num) { if(l <= edge[num].left && r >= edge[num].right) { return edge[num].value; } if(r <= edge[num].mid) return query(l,r,num*2); else if(l >= edge[num].mid + 1) return query(l,r,num *2+1); else { return query(l,edge[num].mid,num*2) + query(edge[num].mid+1,r,num*2+1); } } /* void insert(int l,int r ,int num) //l,r为插入段的左右端点,num为当前结点的下标 { if(l <= edge[num].left && r >= edge[num].right) //插入段完全大于当前段 { edge[num].cover = 1; return ; } if(r <= edge[num].mid) //插入段被当前段的左子树包含 insert(l,r,2 * num); else if(l >= edge[num].mid + 1)//插入段被当前段的右子树包含 insert(l,r,2*num+1); else { insert(l,edge[num].mid, 2* num); insert(edge[num].mid+1,r,2*num+1); } } */ int main() { int t,i,j; int casee = 1; char str[10]; cin >> t; while(t--) { printf("Case %d:\n",casee++); int n,a; cin >> n; build(1,n,1); for(i=1; i<=n; i++) { scanf("%d",&a); update(1,i,a); } while(scanf("%s",str)) { if(str[0] == 'E') break; if(str[0] == 'Q') { scanf("%d%d",&i,&j); printf("%d\n",query(i,j,1)); } if(str[0] == 'A') { scanf("%d%d",&i,&j); update(1,i,j); } if(str[0] == 'S') { scanf("%d%d",&i,&j); update(1,i,-j); } } } return 0; }