59
我就没见过比这道题还狗的题,在OJ上交得看运气,运气好AC,运气不好TLE,真狗啊。
这是我花了接近一天的时间搞明白的线段树,但为什么要用线段树我还是没有搞明白,以后再去研究吧。
线段树自从他出生就和二分法离不开。这里我又学习了C++的cin的一个小用法,就是在接受字符串时,cin遇到
空格就停止,要记住。线段树主要有以下几个操作,建立线段树,更新线段树,搜索线段树。至于网上说的那些
在区间求最值,求区间和啊什么的,那都是在建立线段树的时侯要注意的操作。我发现啊,在这三个操作里边,
都是从ROOT开始操作,这个很好理解的。在区间求和这个函数里,递归结束条件需要注意,自己可以模拟一下加深
理解,算了,写也写不明白,代码一看全都明了。
#include<iostream> #include<stdlib.h> #include<string.h> #include<stdlib.h> using namespace std; struct node { int a; int b; int sum; }t[140000]; int r[50010],SUM=0; void make(int x,int y,int num) { if(x==y) { t[num].a=t[num].b=x; t[num].sum=r[x]; return ; } t[num].a=x; t[num].b=y; int mid=(x+y)/2; make(x,mid,num*2); make(mid+1,y,num*2+1); t[num].sum=t[num*2].sum+t[num*2+1].sum; } void query(int x,int y,int num) { if(x<=t[num].a && y>=t[num].b) { SUM+=t[num].sum; return ; } int mid=(t[num].a+t[num].b)/2; if(y<=mid) //说明搜索区间在左子树 { query(x,y,num*2); } else if(x>mid) { query(x,y,num*2+1); } else { query(x,y,num*2); query(x,y,num*2+1); } } void add(int x,int decrease,int num) //decrease代表增加的值,num默认从1开始 { t[num].sum+=decrease; if(x==t[num].a && x==t[num].b) return ; int mid=(t[num].a+t[num].b)/2; if(x<=mid) { add(x,decrease,num*2); } else { add(x,decrease,num*2+1); } } void sub(int x,int increase,int num) { t[num].sum-=increase; if(x==t[num].a && x==t[num].b) return ; int mid=(t[num].a + t[num].b)/2; if(x<=mid) { sub(x,increase,num*2); } else { sub(x,increase,num*2+1); } } int main() { int T,i,j,re=1; cin>>T; char ch[6]; while(T--) { printf("Case %d:\n",re++); int n; cin>>n; for(i=1;i<=n;i++) cin>>r[i]; make(1,n,1); while(cin>>ch) { if(strcmp(ch,"End")==0) break; if(strcmp(ch,"Query")==0) { SUM=0; int a,b; cin>>a>>b; query(a,b,1); cout<<SUM<<endl; } else if(strcmp(ch,"Add")==0) { int a,b; cin>>a>>b; add(a,b,1); } else { int a,b; cin>>a>>b; sub(a,b,1); } } } return 0; }