<tree>平衡樹

• 需要包含頭文件<ext/pb_ds/tree_policy.hpp>和<ext/pb_ds/assoc_container.hpp>

  或者直接#include <bits/extc++.h>

• 創(chuàng)建下標(biāo)和數(shù)值正相關(guān)平衡樹的基本方法：（負(fù)相關(guān)只需要把less改成greater）

  tree<double, null_type, less<double>, rb_tree_tag, tree_order_statistics_node_update> T;

  //對于：

  • tree<double, null_type, greater, rb_tree_tag, tree_order_statistics_node_update> T;

  //這個東西是有一點長
  //第一個參數(shù)是數(shù)據(jù)類型
  //第二個要填null_type，低版本編譯器填null_mapped_type
  //第三個填比較函數(shù) std::greater<> or std::less<> or cmp
  //第四個填樹的類型，有rb_tree_tag紅黑樹和splay_tree_tag
  //第五個是為了支持查詢第k大和排名的一個參數(shù)
  //tree_order_statistics_node_update

• 基本操作有

T.insert(); //插入一個數(shù)x T.erase(T.lower_bound(x)); //刪除一個數(shù)x T.order_of_key(x); //查詢x在樹中的下標(biāo)（排序后的下標(biāo)） *T.find_by_order(k); //查詢下標(biāo)為k的數(shù)，返回它的迭代器

• 因為這種tree和set一樣不會保存相同的元素，所以在使用時我們可以加上i * 1e-6（其中i是輸入時的下標(biāo)or操作序，唯一就行）

#include<bits/stdc++.h> using namespace std;#include <bits/extc++.h> // #include <ext/pb_ds/tree_policy.hpp> // #include <ext/pb_ds/assoc_container.hpp> using namespace __gnu_pbds;// tree<double, null_mapped_type, greater<double>, rb_tree_tag, tree_order_statistics_node_update> T; tree<double, null_type, less<double>, rb_tree_tag, tree_order_statistics_node_update> T;int main() {// //freopen("3369.in", "r", stdin);//洛谷// //freopen("3369.out", "w", stdout);int q, opt, x;scanf("%d", &q);for (int i = 1; i <= q; ++ i){scanf("%d%d", &opt, &x);if(opt == 1) T.insert(x + i * 1e-6);//插入一個數(shù)if(opt == 2) T.erase(T.lower_bound(x));//刪除一個數(shù)if(opt == 3) printf("%d\n", (int)T.order_of_key(x) + 1);//查詢一個數(shù)的排名if(opt == 4) printf("%d\n", (int)*T.find_by_order(x - 1));//查詢第k小的數(shù) 返回的是一個迭代器 這里k是從0開始算的，意思是最小的數(shù)是第0小的if(opt == 5) printf("%d\n", (int)round(*(-- T.lower_bound(x))));//查詢一個數(shù)的前驅(qū)if(opt == 6) printf("%d\n", (int)round(*T.lower_bound(x + 1)));//查詢一個數(shù)的后繼}return 0; }

總結(jié)

以上是生活随笔為你收集整理的【算法板子】普通平衡树（洛谷P3369）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。