普通平衡树的2种stl实现方法

以luogu上普通平衡树一题为例

vector

#include<bits/stdc++.h>
using namespace std;
#define loop(i,start,end) for(register int i=start;i<=end;++i)
#define anti_loop(i,start,end) for(register int i=start;i>=end;--i)
#define clean(arry,num) memset(arry,num,sizeof(arry))
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a<b)?a:b)
#define ll long long
template<typename T>void read(T &x){
    x=0;char r=getchar();T neg=1;
    while(r>'9'||r<'0'){if(r=='-')neg=-1;r=getchar();}
    while(r>='0'&&r<='9'){x=(x<<1)+(x<<3)+r-'0';r=getchar();}
    x*=neg;
}
int n;
int main(){
	read(n);
	register int op,x;
	vector<int>vec;
	loop(i,1,n){
		read(op);
		read(x);
		if(op==1)
			vec.insert(lower_bound(vec.begin(),vec.end(),x),x);
        else if(op==2)
            vec.erase(lower_bound(vec.begin(),vec.end(),x));
        else if(op==3)
            printf("%d\n",lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1);
        else if(op==4)
            printf("%d\n",vec[x-1]);
        else if(op==5)
            printf("%d\n",vec[lower_bound(vec.begin(),vec.end(),x)-vec.begin()-1]);
        else
            printf("%d\n",vec[upper_bound(vec.begin(),vec.end(),x)-vec.begin()]);
	}
	return 0;
}

0.begin,end

• 返回值：a.begin(),返回a的首地址，a.end(),返回a的尾地址+1

1.lower_bound——logn内查找首个大于等于x的值的地址

• 参数：lower_bound（起始位置，结束位置+1）
  lower_bound(vec.begin(),vec.end(),x)

2.upper_bound——logn内查找首个大于x的值的地址

• 参数:upper_bound（起始位置，结束位置+1）
  upper_bound(vec.begin(),vec.end(),x)

3.insert——插入

• 参数：a.insert（位置，值）
  vec.insert(lower_bound(vec.begin(),vec.end(),x),x)

4.erase——清除

• 参数：a.erase(位置)
  vec.erase(lower_bound(vec.begin(),vec.end(),x))

5.调用

• a[x],返回x位置的值（数组）

multiset

#include<bits/stdc++.h>
using namespace std;
#define loop(i,start,end) for(register int i=start;i<=end;++i)
#define anti_loop(i,start,end) for(register int i=start;i>=end;--i)
#define clean(arry,num) memset(arry,num,sizeof(arry))
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a<b)?a:b)
#define ll long long
template<typename T>void read(T &x){
    x=0;char r=getchar();T neg=1;
    while(r>'9'||r<'0'){if(r=='-')neg=-1;r=getchar();}
    while(r>='0'&&r<='9'){x=(x<<1)+(x<<3)+r-'0';r=getchar();}
    x*=neg;
}
multiset<int>M;
int n;
int main(){
    read(n);
    register int op,x;
    loop(i,1,n){
        read(op);
        read(x);
        if(op==1)
            M.insert(x);
        else if(op==2)
            M.erase(M.lower_bound(x));
        else if(op==3)
            printf("%d\n",distance(M.begin(),M.lower_bound(x))+1);
        else if(op==4){
            multiset<int>::iterator t;
            t=M.begin();
            advance(t,x-1);
            printf("%d\n",*t);
        }
        else if(op==5)
            printf("%d\n",*--M.lower_bound(x));
        else if(op==6)
            printf("%d\n",*M.upper_bound(x));
    }
    return 0;
}

0. begin,end

• 返回值：a.begin(),返回a的首地址，a.end(),返回a的尾地址+1

1.lower_bound——logn内查找首个大于等于x的值的地址

• 参数：a.lower_bound（值）
  M.lower_bound(x)

2.upper_bound——logn内查找首个大于x的值的地址

• 参数:a.upper_bound（值）
  M.upper_bound(x)

3.insert——插入

• 参数：a.insert（值）
  M.insert(x)

4.erase——清除

• 参数：a.erase(位置)
  M.erase(M.lower_bound(x))

5.distance——返回两者之间的距离

• 参数：distance(起始,结束)
  distance(M.begin(),M.lower_bound(x))

6.定义迭代器

• 格式：类型 变量名

multiset<int>::iterator  t

7.advance——将某个迭代器位移

• 参数：advance（迭代器，位移）
  advance(t，x-1);

8.返回地址上的值

• 格式：*地址

*M.upper_bound(x)

9.对地址进行位移

• 格式：++/-- 地址

*--M.lower_bound(x)