Treap学习基本入门

Treap标准学习模板

1.treap的基本了解

(1)splay与treap的区别

Splay的旋转操作是将普通节点转到根
而treap是将根转为普通节点

(2)treap的时空复杂度

treap的各项操作时间复杂度均摊为O(logn)。
由于treap的指针写法容易出错，所以通常用数组代替。
通常要有以下几种：
v[]——存放键值
rnd[]——存放随机出的优先级
l[],r[]——左右子树的下标
w[]——相同键值的个数
s[]——以该结点为祖先的节点个数（包括自己）

2.treap基本操作

左旋和右旋

inline void pushup(int k){s[k]=s[l[k]]+s[r[k]]+w[k];}inline void rturn(int &k){int t=l[k];l[k]=r[t];r[t]=k;s[t]=s[k];pushup(k);k=t;}inline void lturn(int &k){int t=r[k];r[k]=l[t];l[t]=k;s[t]=s[k];pushup(k);k=t;}

一定要明白平衡树通过维持树的平衡，来保证各操作的时间复杂度均摊较低。

插入

void ins(int &k,int num){    if(!k)    {k=++tot;v[k]=num;s[k]=w[k]=1;l[k]=r[k]=0;rnd[k]=rand();return;}    s[k]++;    if(v[k]==num)w[k]++;    else if(num<v[k])      {ins(l[k],num);if(rnd[l[k]]<rnd[k])rturn(k);}    else      {ins(r[k],num);if(rnd[r[k]]<rnd[k])lturn(k);}  }

插入操作其实是一个
*按照键值大小找寻插入位置
*插入后，随机优先级
*进行旋转，保证对于优先级来说，树满足堆的性质
的过程

删除

void del(int &k,int num){    if(v[k]==num)    {        if(w[k]>1){w[k]--;s[k]--;}        else if(l[k]*r[k]==0)k=l[k]+r[k];        else if(rnd[l[k]]<rnd[r[k]])         {rturn(k);del(k,num);}        else {lturn(k);del(k,num);}         return;    }    s[k]--;    if(num<v[k])del(l[k],num);else del(r[k],num);}

删除时：
*先找到待删结点
*只有一棵子树，直接用这棵子树代替这个待删结点
*有两棵子树，先将优先级高的那一棵旋转到根，然后递归在另一棵子树中删除结点

以上是最基本的操作，此外还可有寻找第k大，前驱，后继等功能。理解以上代码后，应该可以编写出。

3.treap入门例题

bzoj1503

http://www.lydsy.com/JudgeOnline/problem.php?id=1503

#include<iostream>#include<cstdio>#include<ctime>#include<cstdlib>using namespace std;const int maxn=100020;int s[maxn],l[maxn],r[maxn],v[maxn],tot=0,rnd[maxn],n,m,pre,root,ans; inline void pushup(int k){s[k]=s[l[k]]+s[r[k]]+1;}inline void rturn(int &k){int t=l[k];l[k]=r[t];r[t]=k;s[t]=s[k];pushup(k);k=t;}inline void lturn(int &k){int t=r[k];r[k]=l[t];l[t]=k;s[t]=s[k];pushup(k);k=t;}void ins(int &k,int num){    if(!k)    {k=++tot;v[k]=num;s[k]=1;l[k]=r[k]=0;rnd[k]=rand();return;}  s[k]++;  if(num<v[k])  {ins(l[k],num);if(rnd[l[k]]<rnd[k])rturn(k);}  else    {ins(r[k],num);if(rnd[r[k]]<rnd[k])lturn(k);}  }int del(int &k,int num){    if(k==0)return 0;    if(v[k]<num){      int t=s[l[k]]+1;k=r[k];return t+del(k,num);  }else{    int t=del(l[k],num);s[k]-=t;return t;    }}int find(int k,int x){    if(s[l[k]]+1==x)return v[k]+pre;    else if(s[l[k]]+1<x)return find(r[k],x-s[l[k]]-1);    else return find(l[k],x);}int main(){    srand(1);    scanf("%d%d",&n,&m);    char ch[1];int x;    while(n--){        scanf("%s%d",ch,&x);        if(ch[0]=='I')if(x>=m)ins(root,x-pre);        if(ch[0]=='A')pre+=x;        if(ch[0]=='S'){            pre-=x;ans+=del(root,m-pre);      }      if(ch[0]=='F'){        if(x>s[root])printf("-1\n");        else printf("%d\n",find(root,s[root]-x+1));      }  }  printf("%d\n",ans);  return 0;}

poj2892

http://poj.org/problem?id=2892

#include<iostream>#include<cstdio>#include<ctime>#include<cstdlib>using namespace std;const int maxn=80005;int l[maxn],r[maxn],v[maxn],rnd[maxn],w[maxn],d[maxn];int root,n,m,s,t,tot;inline void rturn(int &k){int t=l[k];l[k]=r[t];r[t]=k;k=t;}inline void lturn(int &k){int t=r[k];r[k]=l[t];l[t]=k;k=t;}void ins(int &k,int num){    if(k==0){        k=++tot;w[k]=1;v[k]=num;l[k]=r[k]=0;rnd[k]=rand();return;  }  if(num==v[k])w[k]++;  else if(num<v[k]){      ins(l[k],num);if(rnd[l[k]]<rnd[k])rturn(k);  }else{      ins(r[k],num);if(rnd[r[k]]<rnd[k])lturn(k);  }}void del(int &k,int num){    if(v[k]==num){        if(w[k]>1){w[k]--;return;}        if(l[k]*r[k]==0)k=l[k]+r[k];        else if(rnd[l[k]]<rnd[r[k]]){            rturn(k);del(k,num);      }else{        lturn(k);del(k,num);      }      return;  }  if(num<v[k])del(l[k],num);else del(r[k],num);}void find(int &k,int num){    if(k==0)return;    if(v[k]>=num&&v[k]<t)t=v[k];    if(v[k]<=num&&v[k]>s)s=v[k];    if(v[k]<num)find(r[k],num);    else find(l[k],num);}int main(){    freopen("input.txt","r",stdin);    srand(1);    scanf("%d%d",&n,&m);    char ch[1];int a;    for(int i=1,j=0;i<=m;i++){        scanf("%s",ch);        if(ch[0]=='D'){            scanf("%d",&a);            ins(root,a);j++;d[j]=a;      }      if(ch[0]=='R'){del(root,d[j]);j--;}      if(ch[0]=='Q'){        scanf("%d",&a);        s=0;t=n+1;        find(root,a);        if(s==a&&t==a)printf("0\n");        else printf("%d\n",t-s-1);      }      for(int i=1;i<=tot;i++){}    }    return 0;}

---

鸣谢提供模板的善良学长ZYF

鸣谢提供题目的善良学长Revain