HDU1823二维线段树+易错点

话说一开始做线段树看到这题想到应该就是所谓的二维线段树，于是就敲了一下，敲完3K多B之后提交TLE之后一直不知道怎么回事于是去找了相关代码来参考，发现查找，更新，建树的思想是一样，但是我SB了，竟然把左右子树的位置还记下来了，因为之前一直用的动态储存就是指针，所以数据结构中必须要有两个子树的指针，但是这次用了静态的数组来储存，所以习惯性的就把左右儿子的信息记录下来了，后来才发现根本不用的因为如果父节点是N,那左儿子的下标必然是2*N,右儿子下标也必然是2*N+1,因此把这两个变量去掉了，并且操作一律用位运算代替，然后提交发现欣喜的WA了，然后想到给的活泼度左值不一定小于右值，身高也是，于是乎很自然就判断了一下，结果还是WA,然后再仔细看，发现建树的时候SB,因为活泼度的值是从0-1000的我却建树的时候用了1-1000然后把1改成0就A了，140ms不算太快也不算太慢！！线段树的确是个神物！！！必须要好好练习！！#include<iostream> using namespace std; const int Max=2200; const int MaxSec=300; //最好是把第二维的数组开在里面比较节省内存 struct Node { int l,r; int Val; }Root[Max][MaxSec]; //建树第二维 void MakeTreeSecond(int Fir,int Sec,int l,int r) { int Mid; Root[Fir][Sec].l=l; Root[Fir][Sec].r=r; Root[Fir][Sec].Val=0; if(l!=r) { Mid=(Root[Fir][Sec].l+Root[Fir][Sec].r)>>1; MakeTreeSecond(Fir,Sec<<1,l,Mid); MakeTreeSecond(Fir,(Sec<<1)+1,Mid+1,r); } } //建树第一维 void MakeTreeFirst(int Fir,int l,int r) { int Mid; Root[Fir][0].l=l; Root[Fir][0].r=r; if(l!=r) { //递归第二维 MakeTreeSecond(Fir,1,100,200); Mid=(Root[Fir][0].l+Root[Fir][0].r)>>1; MakeTreeFirst(Fir<<1,l,Mid); MakeTreeFirst((Fir<<1)+1,Mid+1,r); } else { MakeTreeSecond(Fir,1,100,200); return ; } } //更新第二维 void UpdateSecond(int Fir,int Sec,int l,int r,int Val,int Height) { //判断边界 if(Sec>=MaxSec) return; int Mid=(Root[Fir][Sec].l+Root[Fir][Sec].r)>>1; //如果更新值比当前值大，说明要更新 Root[Fir][Sec].Val=Root[Fir][Sec].Val>Val?Root[Fir][Sec].Val:Val; if(l!=r) { if(Height<=Mid) { UpdateSecond(Fir,Sec<<1,l,Mid,Val,Height); } else if(Height>Mid) { UpdateSecond(Fir,(Sec<<1)+1,Mid+1,r,Val,Height); } } } //更新第一维 void UpdateFirst(int Fir,int l,int r,int Val,int Viv,int Height) { if(Fir>=Max) return ; int Mid=(Root[Fir][0].l+Root[Fir][0].r)>>1; UpdateSecond(Fir,1,100,200,Val,Height); if(Viv<=Mid) { UpdateFirst(Fir<<1,l,Mid,Val,Viv,Height); } else if(Viv>Mid) { UpdateFirst((Fir<<1)+1,Mid+1,r,Val,Viv,Height); } } //查找第二维 int QuerySecond(int Fir,int Sec,int HeightL,int HeightR) { //如果没找到返回-1 if(Sec>=MaxSec) return -1; int Mid; //完全匹配了就返回值 if(Root[Fir][Sec].l==HeightL&&Root[Fir][Sec].r==HeightR) return Root[Fir][Sec].Val; Mid=(Root[Fir][Sec].l+Root[Fir][Sec].r)>>1; if(HeightR<=Mid) { return QuerySecond(Fir,Sec<<1,HeightL,HeightR); } else if(HeightL>Mid) { return QuerySecond(Fir,(Sec<<1)+1,HeightL,HeightR); } else { int Left=QuerySecond(Fir,Sec<<1,HeightL,Mid); int Right=QuerySecond(Fir,(Sec<<1)+1,Mid+1,HeightR); return Left>Right?Left:Right; } } //查找第一维 int QueryFirst(int Fir,int VivL,int VivR,int HeightL,int HeightR) { if(Fir>=Max) return -1; int Mid; if(VivL==Root[Fir][0].l&&VivR==Root[Fir][0].r) { return QuerySecond(Fir,1,HeightL,HeightR); } Mid=(Root[Fir][0].l+Root[Fir][0].r)>>1; if(VivR<=Mid) { return QueryFirst(Fir<<1,VivL,VivR,HeightL,HeightR); } else if(VivL>Mid) { return QueryFirst((Fir<<1)+1,VivL,VivR,HeightL,HeightR); } else { int Left=QueryFirst(Fir<<1,VivL,Mid,HeightL,HeightR); int Right=QueryFirst((Fir<<1)+1,Mid+1,VivR,HeightL,HeightR); return Left>Right?Left:Right; } } int main() { int n; char ch; while(scanf("%d",&n)!=EOF&&n) { MakeTreeFirst(1,0,1000); getchar(); while(n--) { scanf("%c",&ch); if(ch=='I') { int Height; double Viv,Val; scanf("%d%lf%lf",&Height,&Viv,&Val); Viv*=10; Val*=10; getchar(); int _Viv=Viv; int _Val=Val; UpdateFirst(1,1,1000,_Val,_Viv,Height); } else { double VivL,Res,VivR; int HeightL,HeightR; scanf("%d%d%lf%lf",&HeightL,&HeightR,&VivL,&VivR); getchar(); VivL*=10; VivR*=10; int _VivL=VivL,_VivR=VivR,_HeightL=HeightL,_HeightR=HeightR; //易错点 if(_VivL>_VivR) { _VivL=_VivL^_VivR; _VivR=_VivL^_VivR; _VivL=_VivL^_VivR; } if(_HeightL>_HeightR) { _HeightL=_HeightL^_HeightR; _HeightR=_HeightL^_HeightR; _HeightL=_HeightL^_HeightR; } Res=QueryFirst(1,_VivL,_VivR,_HeightL,_HeightR); if(Res<=0) printf("-1/n"); else printf("%.1lf/n",Res/10.0); } } } return 1; }