zoukankan      html  css  js  c++  java
  • 平衡树板子

    突然发现没存过平衡树板子,趁这几天在复习平衡树,存一下还热乎的未封装版本

    $ m FHQ Treap$
    //12252024832524
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    #define TT template<typename T>
    using namespace std; 
    
    typedef long long LL;
    const int MAXN = 100005;
    const int MOD = 998244353;
    int n;
    
    LL Read()
    {
    	LL x = 0,f = 1;char c = getchar();
    	while(c > '9' || c < '0'){if(c == '-')f = -1;c = getchar();}
    	while(c >= '0' && c <= '9'){x = (x*10) + (c^48);c = getchar();}
    	return x * f;
    }
    TT void Put1(T x)
    {
    	if(x > 9) Put1(x/10);
    	putchar(x%10^48);
    }
    TT void Put(T x,char c = -1)
    {
    	if(x < 0) putchar('-'),x = -x;
    	Put1(x); if(c >= 0) putchar(c);
    }
    TT T Max(T x,T y){return x > y ? x : y;}
    TT T Min(T x,T y){return x < y ? x : y;}
    TT T Abs(T x){return x < 0 ? -x : x;}
    
    int tot,rt;
    struct FHQ_Treap
    {
    	int l,r,val,rd,siz;
    }t[MAXN];
    void up(int x){t[x].siz = t[t[x].l].siz + t[t[x].r].siz + 1;}
    void split(int now,int val,int &x,int &y)
    {
    	if(!now) {x = y = 0;return;}
    	else
    	{
    		if(t[now].val <= val) x = now,split(t[now].r,val,t[now].r,y);
    		else y = now,split(t[now].l,val,x,t[now].l);
    		up(now);
    	}
    }
    int mge(int x,int y)
    {
    	if(!x || !y) return x|y;
    	if(t[x].rd < t[y].rd) {t[x].r = mge(t[x].r,y);up(x);return x;}
    	else {t[y].l = mge(x,t[y].l);up(y);return y;}
    }
    int newnode(int val)
    {
    	t[++tot] = {0,0,val,rand(),1};
    	return tot;
    }
    int x,y,z;
    void ins(int val)
    {
    //	printf("ins ins ins
    ");
    	split(rt,val-1,x,y);
    //	printf("ins %d %d
    ",x,y);
    	rt = mge(x,mge(newnode(val),y));
    }
    void del(int val)
    {
    	split(rt,val,x,z);
    	split(x,val-1,x,y);
    	rt = mge(mge(x,mge(t[y].l,t[y].r)),z);
    }
    int qrk(int val)
    {
    	split(rt,val-1,x,y);
    	int ret = t[x].siz+1;
    	rt = mge(x,y);
    	return ret;
    }
    int qval(int rk)
    {
    	x = rt;
    	while(1)
    	{
    		if(t[t[x].l].siz+1 == rk) return t[x].val;
    		else if(t[t[x].l].siz >= rk) x = t[x].l;
    		else rk -= t[t[x].l].siz+1,x = t[x].r;
    	}
    }
    int pre(int val)
    {
    	split(rt,val-1,x,y);
    	int now = x;
    	while(t[now].r) now = t[now].r;
    	rt = mge(x,y);
    	return t[now].val;
    }
    int suf(int val)
    {
    	split(rt,val,x,y);
    	int now = y;
    	while(t[now].l) now = t[now].l;
    	rt = mge(x,y);
    	return t[now].val;
    }
    void dfs(int now)
    {
    	if(!now) return;
    //	printf("dfs %d %d %d %d
    ",now,t[now].val,t[now].l,t[now].r);
    	dfs(t[now].l);
    	dfs(t[now].r);
    }
    
    int main()
    {
    //	freopen(".in","r",stdin);
    //	freopen(".out","w",stdout);
    	srand(42523);
    	for(int T = Read(); T ;-- T)
    	{
    		int opt = Read();
    		if(opt == 1) ins(Read());
    		else if(opt == 2) del(Read());
    		else if(opt == 3) Put(qrk(Read()),'
    ');
    		else if(opt == 4) Put(qval(Read()),'
    ');
    		else if(opt == 5) Put(pre(Read()),'
    ');
    		else if(opt == 6) Put(suf(Read()),'
    ');
    	}
    	return 0;
    }
    
    $ m Splay$
    //12252024832524
    #include <cstdio>
    #include <cstring>
    #include <algorithm>
    #define TT template<typename T>
    using namespace std; 
    
    typedef long long LL;
    const int MAXN = 100005;
    const int INF = 0x3f3f3f3f;
    int n;
    
    LL Read()
    {
    	LL x = 0,f = 1;char c = getchar();
    	while(c > '9' || c < '0'){if(c == '-')f = -1;c = getchar();}
    	while(c >= '0' && c <= '9'){x = (x*10) + (c^48);c = getchar();}
    	return x * f;
    }
    TT void Put1(T x)
    {
    	if(x > 9) Put1(x/10);
    	putchar(x%10^48);
    }
    TT void Put(T x,char c = -1)
    {
    	if(x < 0) putchar('-'),x = -x;
    	Put1(x); if(c >= 0) putchar(c);
    }
    TT T Max(T x,T y){return x > y ? x : y;}
    TT T Min(T x,T y){return x < y ? x : y;}
    TT T Abs(T x){return x < 0 ? -x : x;}
    
    int tot,rt;
    struct Splay
    {
    	int ch[2],f,siz,cnt,val;
    }t[MAXN];
    void up(int x){t[x].siz = t[t[x].ch[0]].siz + t[t[x].ch[1]].siz + t[x].cnt;}
    int newnode(int val,int fa)
    {
    	t[++tot] = Splay{0,0,fa,1,1,val};
    	if(fa) t[fa].ch[val > t[fa].val] = tot;
    	return tot;
    }
    bool ck(int x){return t[t[x].f].ch[1] == x;}
    void lk(int u,int fa,bool d){t[u].f = fa; if(fa) t[fa].ch[d] = u;}
    void rotate(int x)
    {
    	int fa = t[x].f,d = ck(x);
    	lk(t[x].ch[d^1],fa,d);
    	lk(x,t[fa].f,ck(fa));
    	lk(fa,x,d^1);
    	up(fa); up(x);
    }
    void splay(int x,int to = 0)
    {
    	while(t[x].f^to)
    	{
    		if(t[t[x].f].f^to) 
    		{
    			if(ck(x) == ck(t[x].f)) rotate(t[x].f);
    			else rotate(x);
    		}
    		rotate(x);
    	}
    	if(!to) rt = x;//attention!
    }
    void ins(int val)
    {
    	int x = rt,fa = 0;
    	while(x && (t[x].val^val)) fa = x,x = t[x].ch[val > t[x].val];
    	if(x) ++t[x].cnt,++t[x].siz;
    	else x = newnode(val,fa);
    	splay(x);
    }
    void tort(int val)
    {
    	int x = rt;
    	while(t[x].ch[val > t[x].val] && (t[x].val^val)) x = t[x].ch[val > t[x].val];
    	splay(x);
    }
    int qrk(int val)//latest 
    {
    	tort(val);
    	if(t[rt].val >= val) return t[t[rt].ch[0]].siz;//sentry!
    	return t[rt].siz;
    }
    int qval(int rk)
    {
    	int x = rt; ++rk;//sentry!
    	while(1)
    	{
    		if(t[t[x].ch[0]].siz >= rk) x = t[x].ch[0];
    		else if(t[x].cnt+t[t[x].ch[0]].siz < rk) rk -= t[x].cnt+t[t[x].ch[0]].siz,x = t[x].ch[1];
    		else return t[x].val;
    	}
    }
    int pre(int val)
    {
    	tort(val);
    	if(t[rt].val < val) return rt;
    	int x = t[rt].ch[0];
    	while(t[x].ch[1]) x = t[x].ch[1];
    	return x;
    }
    int suf(int val)
    {
    	tort(val);
    	if(t[rt].val > val) return rt;
    	int x = t[rt].ch[1];
    	while(t[x].ch[0]) x = t[x].ch[0];
    	return x;
    }
    void del(int val)
    {
    	int p = pre(val),s = suf(val);
    	splay(p); splay(s,p);
    	if(t[t[s].ch[0]].cnt > 1) --t[t[s].ch[0]].cnt,--t[t[s].ch[0]].siz;
    	else t[s].ch[0] = 0;
    }
    
    int main()
    {
    //	freopen(".in","r",stdin);
    //	freopen(".out","w",stdout);
    	ins(-INF); ins(INF);
    	for(int T = Read(); T ;-- T)
    	{
    		int opt = Read();
    		if(opt == 1) ins(Read());
    		else if(opt == 2) del(Read());
    		else if(opt == 3) Put(qrk(Read()),'
    ');
    		else if(opt == 4) Put(qval(Read()),'
    ');
    		else if(opt == 5) Put(t[pre(Read())].val,'
    ');
    		else if(opt == 6) Put(t[suf(Read())].val,'
    ');
    	}
    	return 0;
    }
    
  • 相关阅读:
    第五次站立会议
    第四次站立会议
    第三次晚间站立总结会议
    易校小程序典型用户需求分析
    第三次站立会议
    第二次晚间站立总结会议
    第二次站立会议
    第一次晚间站立总结会议
    MyBatis注解
    延迟加载与缓存
  • 原文地址:https://www.cnblogs.com/PPLPPL/p/15223469.html
Copyright © 2011-2022 走看看