HDU 1166 敌兵布阵

我是用线段树解决的，也可以用树状数组解决。

 

以前看别人的代码写过一个有离散化的线段树，现在印象模糊了，就在拾起来线段树。

 

build（）建树函数

 

insert（）修改cnt的函数，递归时的每一段都包含该段，所以他们的cnt都一起改变。

 

query（）查询函数，一开始直接返回的是区间[i,i+1]的值。完全没抛弃了线段树的精髓。ORZ。。

 

#include <iostream> #define MAX 50000 using namespace std; struct node{ int l,r,mid,cnt; }tree[MAX*4]; void build(int l,int r,int num) { tree[num].l=l; tree[num].r=r; tree[num].mid=(l+r)/2; tree[num].cnt=0; if(l+1!=r) { int mid=(l+r)/2; build(l,mid,num<<1); build(mid,r,(num<<1)+1); } } void insert(int l,int r,int num,int val) { tree[num].cnt=tree[num].cnt+val; if(tree[num].l==l&&tree[num].r==r) return; if(r<=tree[num].mid) insert(l,r,num<<1,val); else if(l>=tree[num].mid) insert(l,r,(num<<1)+1,val); else { insert(l,tree[num].mid,num<<1,val); insert(tree[num].mid,r,(num<<1)+1,val); } } int query(int l,int r,int num) { if(tree[num].l==l&&tree[num].r==r) return tree[num].cnt; if(r<=tree[num].mid) query(l,r,num<<1); else if(l>=tree[num].mid) query(l,r,(num<<1)+1); else return query(l,tree[num].mid,num<<1)+query(tree[num].mid,r,(num<<1)+1); } int main() { int t,n,i,j,ai,pos,ll,rr; char ch[10]; scanf("%d",&t); for(i=1;i<=t;++i) { printf("Case %d:/n",i); scanf(" %d",&n); build(1,n+1,1); for(j=1;j<=n;++j) { scanf(" %d",&ai); insert(j,j+1,1,ai); } while(1) { scanf(" %s",&ch); if(ch[0]=='A') { scanf(" %d %d",&pos,&ai); insert(pos,pos+1,1,ai); } else if(ch[0]=='S') { scanf(" %d %d",&pos,&ai); insert(pos,pos+1,1,-ai); } else if(ch[0]=='Q') { scanf(" %d %d",&ll,&rr); printf("%d/n",query(ll,rr+1,1)); } else if(ch[0]=='E') break; } } return 0; }