hdu 1166 敵兵佈陣(線段樹點區)
阿新 • • 發佈:2019-02-04
題目大意: 給出初始化的區間值,然後有三種詢問
Query a b 詢問區間[a,b]值的總和
Add a b 第a個元素的值加b
Sub a b 第a個元素的值減b
解題思路: 線段樹 更新:單點增減 詢問:區間和
每次更新在結點儲存左右子樹值的和,查詢時就不需要查到最低,實現區間查詢
更新時間複雜度O(logN),查詢時間複雜度O(logN)
程式碼:
注:原創文章,轉載請註明出處#include <stdio.h> #include <stdlib.h> #include <string.h> #define MAX 100000 #define MID(a,b) (a+b)>>1 #define L(a) a<<1 #define R(a) (a<<1)+1 typedef struct snode{ int num,left,right; }Node; int num[MAX]; Node Tree[MAX<<1]; void Init() { memset(Tree,0,sizeof(Tree)); } void Build(int t,int l,int r) //以t為根結點,建立左子樹為l,右子樹為r的線段樹 { int mid; Tree[t].left=l,Tree[t].right=r; if(Tree[t].left==Tree[t].right) { Tree[t].num=num[l]; return ; } mid=MID(Tree[t].left,Tree[t].right); Build(L(t),l,mid); Build(R(t),mid+1,r); Tree[t].num=Tree[L(t)].num+Tree[R(t)].num; } void Insert(int t,int l,int r,int n) //向t為根結點,左子樹為l,右子樹為r的結點加上值n { int mid; if(Tree[t].left==l&&Tree[t].right==r) { Tree[t].num+=n; return ; } mid=MID(Tree[t].left,Tree[t].right); if(r<=mid) Insert(L(t),l,r,n); else if(l>=mid) Insert(R(t),l,r,n); else { Insert(L(t),l,mid,n); Insert(R(t),mid+1,r,n); } Tree[t].num+=n; } int Query(int t,int l,int r) //查詢根結點為t,左子樹為l,右子樹為r的結點的值 { int mid; if(Tree[t].left==l&&Tree[t].right==r) { return Tree[t].num; } mid=MID(Tree[t].left,Tree[t].right); if(l>mid) //*** return Query(R(t),l,r); else if(r<=mid) //*** return Query(L(t),l,r); else { int a,b; a=Query(L(t),l,mid); b=Query(R(t),mid+1,r); return a+b; } } int main() { char ch[10]; int t,n,i,i1,a,b; scanf("%d",&t); for(i1=1;i1<=t;i1++) { Init(); //初始化 scanf("%d",&n); for(i=1;i<=n;i++) scanf("%d",&num[i]); Build(1,1,n); //以1為根結點建立線段樹 printf("Case %d:\n",i1); while(scanf("%s",ch)&&strcmp(ch,"End")!=0) { scanf("%d%d",&a,&b); if(strcmp(ch,"Add")==0) //第a個元素加b Insert(1,a,a,b); else if(strcmp(ch,"Sub")==0) //第a個元素減b Insert(1,a,a,-b); else printf("%d\n",Query(1,a,b)); //查詢[a,b]區間值的總和 } } return 0; }