首页 > 代码库 > hdu 1754 splay tree伸展树 初战(单点更新,区间属性查询)
hdu 1754 splay tree伸展树 初战(单点更新,区间属性查询)
题意:与区间查询点更新,点有20W个,询问区间的最大值。曾经用线段树,1000+ms,今天的伸展树,890没ms,差不多。
第一次学习伸展树,一共花了2个单位时间,感觉伸展树真很有用,也很好玩。现在只学了一点点。切个点更新试试。
大致思路:用编号(数组)作为树的键值建树,每插一个数,沿路节点更新最大值(每个结点有一个附加信息标记以之为子树的树所有点的最大值)。所以,查询时【i,j】,只要把i-1伸展到树根,把j+1伸展到I-1下面,那么j+1的左子树就是要的区间了!查该子树根值信息即可(特判端点)!同理,更新操作,只要把待更新的伸展到根,然后更新它,同时维护一下信息即可。
#include<iostream> #include<cstdio> using namespace std; const int maxx=200010; int a[maxx];int child[maxx][2];int fa[maxx];int maxo[maxx]; //a[i]是原来数组值,child左右孩子(0,1),maxo【i】是以序号i为根的子树(包括自身)的最大值。 int root=0; void inline maintain(int n) //点被修改后,该点值的维护 { maxo[n]=maxo[n]>a[n]?maxo[n]:a[n]; } void inline rotate(int x,int f) //f=1右旋,f=0左旋 { int y=fa[x]; maxo[x]=maxo[y]; //最大值的更新 maxo[y]=maxo[child[x][f]]>maxo[child[y][f]]?maxo[child[x][f]]:maxo[child[y][f]]; maintain(y); child[y][!f]=child[x][f]; //三次的重连线,注意顺序。 fa[child[x][f]]=y; if(fa[y]) { if(y==child[fa[y]][0]) child[fa[y]][0]=x; else child[fa[y]][1]=x; } else { root=x; } fa[x]=fa[y]; child[x][f]=y; fa[y]=x; } void splay(int n,int goal) //把序号为i的点转到点goal下面的孩子。 { while(fa[n]!=goal) //一直左右旋即可 { int y=fa[n]; rotate(n,child[y][0]==n?1:0); } } void inline insert(int n) //插入来建树 { int temp=root; if(root==0) //根节点 { root=n; maxo[n]=n; return ; } else { while(1) { maxo[temp]=maxo[temp]<a[n]?a[n]:maxo[temp];//插入时候维护最大值 if(n<temp) //左边 { if(child[temp][0]==0) { child[temp][0]=n; fa[n]=temp; maxo[n]=a[n]; splay(n,0); //注意这里要伸展,否则建树就是一般的排序二叉树,会超时 return ; } temp=child[temp][0]; } else //右边 { if(child[temp][1]==0) { child[temp][1]=n; fa[n]=temp; maxo[n]=a[n]; splay(n,0); return ; } temp=child[temp][1]; } } } } void update(int n,int x) //更新 ,把序号为n的值更新为x { splay(n,0); a[n]=x; maxo[n]=maxo[child[n][0]]>maxo[child[n][1]]?maxo[child[n][0]]:maxo[child[n][1]]; maintain(n); } void clear() //初始化 { root=0; for(int i=0;i<maxx;i++) { maxo[i]=child[i][0]=child[i][1]=fa[i]=0; } } int main() { int n,m; while(~scanf("%d%d",&n,&m)) { clear(); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); insert(i); } char q;int i,j; while(m--) { getchar(); scanf("%c%d%d",&q,&i,&j); if(q=='Q') { int ans=0; if(i==1&&j!=n) //区间端点的特判 { splay(j+1,0); ans=maxo[child[j+1][0]]; } else if(i!=1&&j==n) { splay(i-1,0); ans=maxo[child[i-1][1]]; } else if(i==1&&j==n) { ans=maxo[root]; } else { splay(i-1,0); splay(j+1,i-1); ans=maxo[child[j+1][0]]; } printf("%d\n",ans); } else { update(i,j); } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。