[Splay][學習筆記]

来源:https://www.cnblogs.com/wxyww/archive/2018/12/09/10090186.html
-Advertisement-
Play Games

胡扯 因為先學習的treap,而splay與treap中有許多共性,所以會有很多地方不會講的很細緻。關於treap和平衡樹可以參考這篇博客 ...


胡扯

因為先學習的treap,而splay與treap中有許多共性,所以會有很多地方不會講的很細緻。關於treap和平衡樹可以參考這篇博客

關於splay

splay,又叫伸展樹,是一種二叉排序樹,它能在O(log n)內完成插入、查找和刪除操作。它由Daniel Sleator和Robert Tarjan創造。伸展樹是一種自調整形式的二叉查找樹,它會沿著從某個節點到樹根之間的路徑,通過一系列的旋轉把這個節點搬移到樹根去。

splay與其他平衡樹相比功能更加強大,可以處理區間問題。可以說其他平衡樹可以做的splay幾乎都能做。所以很多大佬都說平衡樹會寫splay就好了。唯一的缺點可能就是常數要比treap大。

定義

struct node {
    int ch[2],val,siz,cnt,pre;
}TR[N];

ch[0/1]分別為當前節點的兩個兒子。val為當前節點的值。siz為以當前節點為根的子樹大小,cnt為當前節點的值出現的次數。pre為當前節點的父親節點

旋轉

splay也是通過旋轉來保持平衡的。spaly的旋轉也是挺易懂的

如圖,現在我們把4旋轉到2這個位置。也就是說要把4號節點旋轉上去。

第一步:將2-4這條邊斷開,將8變為2的右兒子。

第二步:將1-2這條邊斷開,將4變為1的右兒子。

第三步:將2變為4的左兒子

第四步:更新4號節點和2號節點,完成旋轉

void rotate(int cur) {
    int f = getwh(cur),fa = TR[cur].pre,gr = TR[fa].pre;
    TR[gr].ch[getwh(fa)] = cur;
    TR[cur].pre = gr;
    TR[fa].ch[f] = TR[cur].ch[f ^ 1];
    TR[TR[cur].ch[f ^ 1]].pre = fa;
    TR[fa].pre = cur;
    TR[cur].ch[f ^ 1] = fa;
    up(fa);
    up(cur);
}

其中getwh是用來得到當前點是其父親的左兒子還是右兒子,fa是當前點的父親,gr是當前點的爺爺

getwh代碼如下

int getwh(int cur) {
    return TR[TR[cur].pre].ch[1] == cur;
}

伸展

與treap相比,splay多了一種非常重要的操作——伸展操作。

所謂伸展,就是通過一系列旋轉,將一個節點挪到一個想讓他到達的位置(這個位置一般為根)。

splay的伸展總共可以分為三種情況(伸展到底有什麼用後面會提到,現在只需知道他的作用如上即可)。

第一種情況:

如圖,x結點要挪到他的爺爺結點下麵,這種情況只要將x點旋轉一次即可

第二種情況

x結點要挪到他爺爺的節點以上的節點下麵,並且他的爺爺,和他的父親,和他在同一直線上。

啥叫在同一直線上???

如圖,現在g,p,x就在同一直線上,然後要將x轉到右面的情況,只要現將p旋轉上去,然後再講x旋轉上去即可

第三種情況,

x結點要挪到他爺爺的節點以上的節點下麵,並且他的爺爺,和他的父親,和他在同一直線上。

如圖,現在只要先將x旋轉到p的位置,然後再將x旋轉到g的位置即可

PS

經博主實踐證明,第二和第三種情況都可以通過第三種情況的操作方式進行,至於為什麼第二類不如此操作,大概是為了保持樹的平衡。但是反而更慢

綜上所述

我們可以得到伸展的代碼(to為0時就是旋轉成根)

void splay(int cur,int to) {
    while(TR[cur].pre != to) {
        if(TR[TR[cur].pre].pre != to) {
             if(getwh(cur) == getwh(TR[cur].pre)) rotate(TR[cur].pre);
             else rotate(cur);
        }
        rotate(cur);
    }
    if(!to) rt = cur;
}

插入

splay的插入和treap的插入類似。就是不斷地尋找當前點恰當的位置,如果以前已經有了,就將cnt++即可,否則新建一個節點。

最後不要忘記將新插入的節點伸展為根。

void insert(int cur,int val,int lst) {
    if(!cur) {
        cur = ++tot;
        TR[cur].pre = lst;
        TR[cur].siz = TR[cur].cnt = 1;
        TR[cur].val = val;
        TR[lst].ch[val > TR[lst].val] = cur;
        splay(cur,0);
        return;
    }
    TR[cur].siz++;
    if(val == TR[cur].val) {TR[cur].cnt++;return;}
    if(val > TR[cur].val) insert(rs,val,cur);
    else insert(ls,val,cur);
}

合併

合併操作也是treap種所沒有了。splay中的合併主要是為了刪除操作做準備

所謂合併也就是將兩棵子樹合成一棵。兩棵子樹能合併的前提是其中一個中的所有元素大於另一棵的所有元素。

其實很簡單,假如說現在x子樹中的所有元素都小於y子樹中的所有元素,那麼只需找到x種最靠右的(也就是最大的)節點,然後將y變為此節點的右孩子即可。

最後還是要把y節點或者是x子樹中最大的那個節點伸展為根。

void merge(int cur,int y) {
    if(TR[cur].val > TR[y].val) swap(cur,y);
    if(!cur) {
        rt = y;
        return;
    }
    while(rs) cur = rs;
    splay(cur,0);
    rs = y;
    TR[y].pre = cur;
    up(cur);
}

查找節點

這也是一個用來輔助其他操作的操作。作用是查找權值為val的節點。

很簡單,就是在二叉搜索樹上查找操作,如果比當前節點大就去查右子樹,否則查左子樹。和當前節點一樣大了就範圍即可。

int getpos(int cur,int val) {
    int lst;
    while(cur) {
        lst = cur;
        if(TR[cur].val == val) return cur;
        cur = TR[cur].ch[val > TR[cur].val];
    }
    return lst;
}

刪除

有了合併操作,刪除就很好完成了。首先找到要刪除的節點,然後將此節點伸展為根。然後將這個節點的左右子樹合併即可。

void del(int cur,int val) {
    cur = getpos(rt,val);
    if(!cur) return;
    if(TR[cur].val != val) return;
    splay(cur,0);
    if(TR[cur].cnt > 1) {TR[cur].cnt--;TR[cur].siz--;return;}
    TR[ls].pre = TR[rs].pre = 0;
    merge(ls,rs);
}

查詢排名

用查找操作找到當前節點,然後旋轉為根,左子樹的大小+1就是這個節點的排名

查詢第k大

與treap一樣,如果k大於左子樹大小+當前節點個數,就在右子樹中查找k-左子樹大小-當前節點個數。如果k<=左子樹大小,那麼直接在左子樹中查找第k大。否則返回當前點即可。

int kth(int cur,int x) {
    while(cur) {
        if(x <= TR[ls].siz) cur = ls;
        else if(x > TR[ls].siz + TR[cur].cnt) x -= TR[cur].cnt + TR[ls].siz,cur = rs;
        else return TR[cur].val;
    }
    return cur;
}

前驅

找到要查詢的點伸展為根。然後在左子樹中查找最大值即可。

int pred(int cur,int val) {
    cur = getpos(rt,val);
    if(TR[cur].val < val) return TR[cur].val;
    splay(cur,0);
    cur = ls;
    while(rs) cur = rs;
    return TR[cur].val;
}

後繼

找到要查詢的點伸展為根。然後在右子樹中查找最小值即可。

int nex(int cur,int val) {
    cur = getpos(rt,val);
    if(TR[cur].val > val) return TR[cur].val;
    splay(cur,0);
    cur = rs;
    while(ls) cur = ls;
    return TR[cur].val;
}

區間操作

關於splay的區間操作,可以參考這篇博客

完整代碼

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
const int N = 100000 + 100;
#define ls TR[cur].ch[0]
#define rs TR[cur].ch[1]
ll read() {
    ll x = 0,f = 1;char c = getchar();
    while(c < '0' || c > '9') {
        if(c == '-') f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9') {
        x = x * 10 + c -'0';
        c = getchar();
    }
    return x * f;
}
int rt;
struct node {
    int ch[2],val,siz,cnt,pre;
}TR[N];
void up(int cur) {
    TR[cur].siz = TR[ls].siz + TR[rs].siz + TR[cur].cnt;
}
int getwh(int cur) {
    return TR[TR[cur].pre].ch[1] == cur;
}
void rotate(int cur) {
    int f = getwh(cur),fa = TR[cur].pre,gr = TR[fa].pre;
    TR[gr].ch[getwh(fa)] = cur;
    TR[cur].pre = gr;
    TR[fa].ch[f] = TR[cur].ch[f ^ 1];
    TR[TR[cur].ch[f ^ 1]].pre = fa;
    TR[fa].pre = cur;
    TR[cur].ch[f ^ 1] = fa;
    up(fa);
    up(cur);
}
void splay(int cur,int to) {
    while(TR[cur].pre != to) {
        if(TR[TR[cur].pre].pre != to) {
            // if(getwh(cur) == getwh(TR[cur].pre)) rotate(TR[cur].pre);
            // else 
                rotate(cur);
        }
        rotate(cur);
    }
    if(!to) rt = cur;
}
int tot;
void insert(int cur,int val,int lst) {
    if(!cur) {
        cur = ++tot;
        TR[cur].pre = lst;
        TR[cur].siz = TR[cur].cnt = 1;
        TR[cur].val = val;
        TR[lst].ch[val > TR[lst].val] = cur;
        splay(cur,0);
        return;
    }
    TR[cur].siz++;
    if(val == TR[cur].val) {TR[cur].cnt++;return;}
    if(val > TR[cur].val) insert(rs,val,cur);
    else insert(ls,val,cur);
}
void merge(int cur,int y) {
    if(TR[cur].val > TR[y].val) swap(cur,y);
    if(!cur) {
        rt = y;
        return;
    }
    while(rs) cur = rs;
    splay(cur,0);
    rs = y;
    TR[y].pre = cur;
    up(cur);
}
int getpos(int cur,int val) {
    int lst;
    while(cur) {
        lst = cur;
        if(TR[cur].val == val) return cur;
        cur = TR[cur].ch[val > TR[cur].val];
    }
    return lst;
}
void del(int cur,int val) {
    cur = getpos(rt,val);
    if(!cur) return;
    if(TR[cur].val != val) return;
    splay(cur,0);
    if(TR[cur].cnt > 1) {TR[cur].cnt--;TR[cur].siz--;return;}
    TR[ls].pre = TR[rs].pre = 0;
    merge(ls,rs);
}
int Rank(int cur,int val) {
    cur = getpos(rt,val);
    splay(cur,0);
    return TR[ls].siz + 1;
}
int kth(int cur,int x) {
    while(cur) {
        if(x <= TR[ls].siz) cur = ls;
        else if(x > TR[ls].siz + TR[cur].cnt) x -= TR[cur].cnt + TR[ls].siz,cur = rs;
        else return TR[cur].val;
    }
    return cur;
}
int pred(int cur,int val) {
    cur = getpos(rt,val);
    if(TR[cur].val < val) return TR[cur].val;
    splay(cur,0);
    cur = ls;
    while(rs) cur = rs;
    return TR[cur].val;
}
int nex(int cur,int val) {
    cur = getpos(rt,val);
    if(TR[cur].val > val) return TR[cur].val;
    splay(cur,0);
    cur = rs;
    while(ls) cur = ls;
    return TR[cur].val;
}
int main() {
    int n = read();
    while(n--) {
        int opt = read(),x = read();
        if(opt == 1) insert(rt,x,0);
        if(opt == 2) del(rt,x);
        if(opt == 3) printf("%d\n",Rank(rt,x));
        if(opt == 4) printf("%d\n",kth(rt,x));
        if(opt == 5) printf("%d\n",pred(rt,x));
        if(opt == 6) printf("%d\n",nex(rt,x));
    }
    return 0;
}

您的分享是我們最大的動力!

-Advertisement-
Play Games
更多相關文章
  • 最近在開發小程式,個人覺得小程式開發很有意思,體驗也很流暢,很贊!(由於是初版,頁面及體驗還需優化,購物流程是ok的,發現BUG及改進建議歡迎在下方留言,謝謝!) 話不多說,附上部分效果圖!手機預覽打開微信搜索“樂活城市”小程式。(初次接觸小程式的小伙伴們,可以直接去官方給出的小程式API腦補,我這 ...
  • Bootstrap -- 表格樣式、表單佈局 1. 表格的一些樣式 舉例: <!DOCTYPE html> <html> <head> <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/> <title>My Tes ...
  • Python 項目的組織結構 - 包 -- 模塊 類 函數、變數 Python是利用包和模塊來組織一個項目的。 包: 包的物理表現是一個文件夾,但是一個文件夾卻不一定是個包,要想讓個文件夾成為一個包,就必須在這個文件夾下增加個特定的文件 __init__.py __init__.py裡面可以什麼也不 ...
  • springMVC第一個案例---hellospringMVC ...
  • 建造者模式通過將複雜對象逐一拆解成單一的簡單對象,然後通過對簡單對象的創建,最終構建出一個複雜對象。 介紹 在現實世界中,和建造者模式最為相似的是我們到餐廳點餐的流程。在點餐的過程中,我們是不用關係點餐的先後順序,等我們點完後,點餐系統會自動將我們的所有餐品列表和消費情況全部一次性羅列出來,並且最後 ...
  • 單例模式就是指單例類在一定的生命周期內只能有一個對象實例,單例類的創建必須是本身,並能給使用者提供自身。 介紹 在現實世界中,每個生命體都可以被看做是一個單例對象,唯一且具體,具有不可複製性。同樣的,在軟體開發領域中,有時我們需要保證客戶端在當前的客戶機上只能運行一個實例這個時候,我們就應該考慮使用 ...
  • 系統架構設計師-軟體水平考試高級-理論-架構風格。其中涉及架構風格,ABSD,軟體架構評估,軟體產品線,中間件技術,典型應用架構,Web架構設計等。 ...
  • 抽象工廠是基於簡單工廠發展而來的,通過抽象工廠,我們可以創建多種類型的工廠,並且依據具體業務需求而在具體工廠裡面進行任意拼裝組合。 介紹 在現實世界中,汽車製作行業有各種各樣的工廠,每個工廠都需要具有生產輪胎、汽車引擎等部件的能力,但是針對具體的工廠,每個部件的生產又各不相同,所有在軟體開發過程中, ...
一周排行
    -Advertisement-
    Play Games
  • 移動開發(一):使用.NET MAUI開發第一個安卓APP 對於工作多年的C#程式員來說,近來想嘗試開發一款安卓APP,考慮了很久最終選擇使用.NET MAUI這個微軟官方的框架來嘗試體驗開發安卓APP,畢竟是使用Visual Studio開發工具,使用起來也比較的順手,結合微軟官方的教程進行了安卓 ...
  • 前言 QuestPDF 是一個開源 .NET 庫,用於生成 PDF 文檔。使用了C# Fluent API方式可簡化開發、減少錯誤並提高工作效率。利用它可以輕鬆生成 PDF 報告、發票、導出文件等。 項目介紹 QuestPDF 是一個革命性的開源 .NET 庫,它徹底改變了我們生成 PDF 文檔的方 ...
  • 項目地址 項目後端地址: https://github.com/ZyPLJ/ZYTteeHole 項目前端頁面地址: ZyPLJ/TreeHoleVue (github.com) https://github.com/ZyPLJ/TreeHoleVue 目前項目測試訪問地址: http://tree ...
  • 話不多說,直接開乾 一.下載 1.官方鏈接下載: https://www.microsoft.com/zh-cn/sql-server/sql-server-downloads 2.在下載目錄中找到下麵這個小的安裝包 SQL2022-SSEI-Dev.exe,運行開始下載SQL server; 二. ...
  • 前言 隨著物聯網(IoT)技術的迅猛發展,MQTT(消息隊列遙測傳輸)協議憑藉其輕量級和高效性,已成為眾多物聯網應用的首選通信標準。 MQTTnet 作為一個高性能的 .NET 開源庫,為 .NET 平臺上的 MQTT 客戶端與伺服器開發提供了強大的支持。 本文將全面介紹 MQTTnet 的核心功能 ...
  • Serilog支持多種接收器用於日誌存儲,增強器用於添加屬性,LogContext管理動態屬性,支持多種輸出格式包括純文本、JSON及ExpressionTemplate。還提供了自定義格式化選項,適用於不同需求。 ...
  • 目錄簡介獲取 HTML 文檔解析 HTML 文檔測試參考文章 簡介 動態內容網站使用 JavaScript 腳本動態檢索和渲染數據,爬取信息時需要模擬瀏覽器行為,否則獲取到的源碼基本是空的。 本文使用的爬取步驟如下: 使用 Selenium 獲取渲染後的 HTML 文檔 使用 HtmlAgility ...
  • 1.前言 什麼是熱更新 游戲或者軟體更新時,無需重新下載客戶端進行安裝,而是在應用程式啟動的情況下,在內部進行資源或者代碼更新 Unity目前常用熱更新解決方案 HybridCLR,Xlua,ILRuntime等 Unity目前常用資源管理解決方案 AssetBundles,Addressable, ...
  • 本文章主要是在C# ASP.NET Core Web API框架實現向手機發送驗證碼簡訊功能。這裡我選擇是一個互億無線簡訊驗證碼平臺,其實像阿裡雲,騰訊雲上面也可以。 首先我們先去 互億無線 https://www.ihuyi.com/api/sms.html 去註冊一個賬號 註冊完成賬號後,它會送 ...
  • 通過以下方式可以高效,並保證數據同步的可靠性 1.API設計 使用RESTful設計,確保API端點明確,並使用適當的HTTP方法(如POST用於創建,PUT用於更新)。 設計清晰的請求和響應模型,以確保客戶端能夠理解預期格式。 2.數據驗證 在伺服器端進行嚴格的數據驗證,確保接收到的數據符合預期格 ...