http://hihocoder.com/contest/hiho104
以hiho104为例
看着hiho里的说明,写了一个这个东西,先上一个好理解的模板
#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
typedef long long LL;
typedef set<int>::iterator sit;
const LL mod=1e9+7;
const double eps=1e-9;
const int INF=0x3f3f3f3f;
struct Stree{//空节点统一指向0
int fa;
int val;
int l,r;
int siz;
}sp[200010];
int root,cnt;
void pushup(int rt){
sp[rt].siz=sp[sp[rt].l].siz+sp[sp[rt].r].siz+1;
}
void rro(int rt){
int fa=sp[rt].fa;
int rs=sp[rt].r;
int gf=sp[fa].fa;
sp[fa].l=rs;
sp[rs].fa=fa;
if(gf){
if(sp[gf].l==fa)sp[gf].l=rt;
else sp[gf].r=rt;
}else root=rt;
sp[rt].fa=gf;
sp[rt].r=fa;
sp[fa].fa=rt;
pushup(fa);pushup(rt);
}
void lro(int rt){
int fa=sp[rt].fa;
int ls=sp[rt].l;
int gf=sp[fa].fa;
sp[fa].r=ls;
sp[ls].fa=fa;
if(gf){
if(sp[gf].l==fa)sp[gf].l=rt;
else sp[gf].r=rt;
}else root=rt;
sp[rt].fa=gf;
sp[rt].l=fa;
sp[fa].fa=rt;
pushup(fa);pushup(rt);
}
void splay(int x,int y){//把编号为x的节点转到y的儿子节点
while(sp[x].fa!=y){
int fa=sp[x].fa;
if(sp[fa].fa==y){
if(sp[fa].l==x)rro(x);
else lro(x);
}
else {
int gf=sp[fa].fa;
if(sp[fa].l==x&&sp[gf].l==fa){rro(fa);rro(x);}
else if(sp[fa].r==x&&sp[gf].r==fa){lro(fa);lro(x);}
else if(sp[fa].r==x&&sp[gf].l==fa){lro(x);rro(x);}
else {rro(x);lro(x);}
}
}
}
void ist(int &k,int x,int fa){
if(!k){
k=++cnt;
sp[k].val=x;
sp[k].fa=fa;
sp[k].siz=1;
splay(k,0);
return ;
}
if(x<sp[k].val)ist(sp[k].l,x,k);
else ist(sp[k].r,x,k);
}
int pre,nxt;
void fdpre(int rt,int x){//找到第一个小于等于x的数,并保存在pre中
if(!rt)return ;
//printf("%d %d\n",rt,sp[rt].val);
if(sp[rt].val==x){pre=rt;return ;}
if(sp[rt].val<x){pre=rt;fdpre(sp[rt].r,x);}
else {fdpre(sp[rt].l,x);}
}
void fdnxt(int rt,int x){//找到第一个大于等于x的数,并保存在nxt中
if(!rt)return ;
if(sp[rt].val==x){nxt=rt;return ;}
else if(sp[rt].val>x){nxt=rt;fdnxt(sp[rt].l,x);}
else {fdnxt(sp[rt].r,x);}
}
void del(int l,int r){//删除区间[l,r]
fdpre(root,l-1);fdnxt(root,r+1);
splay(pre,0);splay(nxt,pre);
sp[nxt].l=0;
pushup(nxt);pushup(pre);
}
void init(){
cnt=0;root=0;
ist(root,-INF,0);
ist(root,INF,0);
}
void print(int rt){
if(!rt)return ;
print(sp[rt].l);
printf("%d %d %d %d\n",rt,sp[rt].val,sp[rt].l,sp[rt].r);
print(sp[rt].r);
}
char op[2];
int main(void)
{
int n,k,a,b;
while(~scanf("%d",&n)){
init();
while(n--){
scanf("%s",op);
if(op[0]=='I'){
scanf("%d",&k);
ist(root,k,0);
//print(root);
}
else if(op[0]=='Q'){
scanf("%d",&k);
fdpre(root,k);
printf("%d\n",sp[pre].val);
splay(pre,0);
}
else {
scanf("%d%d",&a,&b);
del(a,b);
}
//for(int i=1;i<=7;i++)printf("%d %d %d %d\n",i,sp[i].val,sp[i].l,sp[i].r);
}
}
return 0;
}
果然还是显得太长了,看一看大佬们的写法,再学习一下,打算再压一下
https://blog.csdn.net/qq_30974369/article/details/77587168
看一下这个大佬的讲解
updated:运行效率上没有任何区别,主要是改掉了rotate函数,然后顺便改下面的函数的时候发现很多东西可以省略掉
#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
using namespace std;
typedef long long LL;
typedef set<int>::iterator sit;
const LL mod=1e9+7;
const double eps=1e-9;
const int INF=0x3f3f3f3f;
struct Stree{//空节点统一指向0
int fa,ch[2],val;
int siz;
}sp[200010];
int root,cnt;//根,节点个数
void pushup(int rt){
sp[rt].siz=sp[sp[rt].ch[0]].siz+sp[sp[rt].ch[1]].siz+1;
}
void ro(int rt){
int fa=sp[rt].fa;
int gf=sp[fa].fa;
int sid=sp[fa].ch[1]==rt;
sp[sp[gf].ch[sp[gf].ch[1]==fa]=rt].fa=gf;
sp[sp[fa].ch[sid]=sp[rt].ch[sid^1]].fa=fa;
sp[sp[rt].ch[sid^1]=fa].fa=rt;
if(!gf)root=rt;
pushup(fa);pushup(rt);
}
void splay(int x,int y){//把编号为x的节点转到y的儿子节点
while(sp[x].fa!=y){
int fa=sp[x].fa;
int gf=sp[fa].fa;
if(gf!=y){
if((sp[gf].ch[1]==fa)^(sp[fa].ch[1]==x))ro(x);
else ro(fa);
}
ro(x);
}
}
void ist(int &k,int x,int fa){
if(!k){
k=++cnt;
sp[k].val=x;sp[k].fa=fa;
sp[k].siz=1;splay(k,0);
return ;
}
if(x<sp[k].val)ist(sp[k].ch[0],x,k);
else ist(sp[k].ch[1],x,k);
}
int pre,nxt;
void fdpre(int rt,int x){//找到第一个小于等于x的数,并保存在pre中
if(!rt)return ;
//printf("%d %d\n",rt,sp[rt].val);
if(sp[rt].val==x){pre=rt;return ;}
if(sp[rt].val<x){pre=rt;fdpre(sp[rt].ch[1],x);}
else {fdpre(sp[rt].ch[0],x);}
}
void fdnxt(int rt,int x){//找到第一个大于等于x的数,并保存在nxt中
if(!rt)return ;
if(sp[rt].val==x){nxt=rt;return ;}
else if(sp[rt].val>x){nxt=rt;fdnxt(sp[rt].ch[0],x);}
else {fdnxt(sp[rt].ch[1],x);}
}
void del(int l,int r){//删除区间[l,r]
fdpre(root,l-1);fdnxt(root,r+1);
splay(pre,0);splay(nxt,pre);
sp[nxt].ch[0]=0;
pushup(nxt);pushup(pre);
}
void init(){
cnt=0;root=0;
ist(root,-INF,0);
ist(root,INF,0);
}
void print(int rt){
if(!rt)return ;
print(sp[rt].ch[0]);
printf("%d %d %d %d\n",rt,sp[rt].val,sp[rt].ch[0],sp[rt].ch[1]);
print(sp[rt].ch[1]);
}
char op[2];
int main(void)
{
int n,k,a,b;
while(~scanf("%d",&n)){
init();
while(n--){
scanf("%s",op);
if(op[0]=='I'){
scanf("%d",&k);
ist(root,k,0);
//print(root);
}
else if(op[0]=='Q'){
scanf("%d",&k);
fdpre(root,k);
printf("%d\n",sp[pre].val);
splay(pre,0);
}
else {
scanf("%d%d",&a,&b);
del(a,b);
}
// printf("%d\n",root);
// for(int i=1;i<=7;i++)printf("%d %d %d %d\n",i,sp[i].val,sp[i].ch[0],sp[i].ch[1]);
}
}
return 0;
}
在接下来就是更新用法了。。。挖个坑吧,慢慢填