首页 > 代码库 > hdu-3436-Queue-jumpers-伸展树
hdu-3436-Queue-jumpers-伸展树
5KB的代码。。。250+行。。。就错在一个离散化上,郁闷了好久。。。
RANK就是找出第K位是多少
TOP是将某个人移至队首,对中间区间没有影响
QUERY是某个人的位置
则:
TOP:将目标点旋转至根部,然后删除,最后插入到队首
RANK:通过size查找即可,注意每个点的size是区间长度
QUERY:把该点旋转至根部,左子树的大小+1便是结果
#include<stdio.h> #include<string.h> #include<algorithm> #include<iostream> using namespace std; #define maxn 220000 #define mem(a,b) memset(a,b,sizeof(a)) struct qu { int st; int ed; } cnt[maxn]; int num; int search(int x) { int l=1; int r=num+1; int mid=(l+r)/2; while(l<r) { if(cnt[mid].st<=x)l=mid+1; else r=mid; mid=(l+r)/2; } mid-=1; return mid; } struct list { int x; int st; friend bool operator <(const list &a,const list &b) { return a.x<b.x; } } node[maxn],nodes[maxn]; int pre[maxn],ch[maxn][2],root,tot; int size[maxn]; int val[maxn]; int pos[maxn]; int key[maxn]; int sum[maxn]; int n; void Treaval(int x) { if(x) { Treaval(ch[x][0]); printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size = %2d ,val = %2d , key = %2d \n",x,ch[x][0],ch[x][1],pre[x],size[x],val[x],key[x]); Treaval(ch[x][1]); } } void debug() { printf("debug--%d\n",root); Treaval(root); } //debug专用 void init() { root=tot=0; mem(pre,0); mem(ch,0); mem(size,0); num=1; } void newnode(int &x,int father,int k) { x=++tot; pre[x]=father; val[x]=cnt[k].ed-cnt[k].st+1; size[x]=val[x]; pos[k]=x; key[x]=k; ch[x][0]=ch[x][1]=0; } void push_down(int x) { } void push_up(int x) { size[x]=size[ch[x][0]]+size[ch[x][1]]+val[x]; } void rot(int x,int kind) { int y=pre[x]; ch[y][!kind]=ch[x][kind]; pre[ch[x][kind]]=y; if(pre[y])ch[pre[y]][ch[pre[y]][1]==y]=x; pre[x]=pre[y]; ch[x][kind]=y; pre[y]=x; push_up(y); } void splay(int x,int goal) { while(pre[x]!=goal) { if(pre[pre[x]]==goal) { rot(x,ch[pre[x]][0]==x); } else { int y=pre[x]; int kind=ch[pre[y]][0]==y; if(ch[y][kind]==x) { rot(x,!kind); rot(x,kind); } else { rot(y,kind); rot(x,kind); } } } push_up(x); if(goal==0)root=x; } int getmax(int x) { while(ch[x][1]) { x=ch[x][1]; } return x; } void erase() { int m=getmax(ch[root][0]); splay(m,root); ch[m][1]=ch[root][1]; pre[ch[root][1]]=m; root=m; pre[root]=0; push_up(root); } void insert(int &r,int k,int father) { if(r==0) { newnode(r,father,k); } else { insert(ch[r][0],k,r); push_up(r); } } void top(int x) { int s=search(x); int b=pos[s]; splay(b,0); if(!ch[root][0]||!ch[root][1]) { root=ch[root][0]+ch[root][1]; pre[root]=0; } else erase(); insert(root,s,0); splay(pos[s],0); } void buildtree(int &x,int l,int r,int father) { if(l>r)return ; int mid=(l+r)/2; newnode(x,father,mid); buildtree(ch[x][0],l,mid-1,x); buildtree(ch[x][1],mid+1,r,x); push_up(x); } int get_rank(int r,int k) { int t=size[ch[r][0]]; if(k<=t)return get_rank(ch[r][0],k); else if(k<=t+val[r])return cnt[key[r]].st+(k-t)-1; else return get_rank(ch[r][1],k-t-val[r]); } int get_kth(int x) { int s=search(x); int p=pos[s]; splay(p,0); return size[ch[root][0]]+1; } int main() { int T,_; scanf("%d",&T); for(_=1; _<=T; _++) { init(); int q,i; char str[110]; int x; scanf("%d%d",&n,&q); for(i=1; i<=q; i++) { scanf("%s %d",str,&x); if(strcmp(str,"Top")==0)node[i].st=1; else if(strcmp(str,"Query")==0)node[i].st=2; else node[i].st=3; node[i].x=x; nodes[i]=node[i]; } sort(node+1,node+q+1); node[0].x=0; for(i=1; i<=q; i++) { if(node[i].st==3)continue; if(node[i].x-cnt[num-1].ed>1) { cnt[num].st=cnt[num-1].ed+1; cnt[num++].ed=node[i].x-1; } if(node[i].x-cnt[num-1].ed>0) { cnt[num].st=node[i].x; cnt[num++].ed=node[i].x; } } if(cnt[num-1].ed!=n) { cnt[num].st=cnt[num-1].ed+1; cnt[num].ed=n; } else num--; buildtree(root,1,num,0); printf("Case %d:\n",_); for(i=1; i<=q; i++) { if(nodes[i].st==1) { top(nodes[i].x); } else if(nodes[i].st==2) { cout<<get_kth(nodes[i].x)<<endl; } else { cout<<get_rank(root,nodes[i].x)<<endl; } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。