权值线段树理解及模板

权值线段树,顾名思义就是记录权值的线段树,普通的线段树直接以坐标为l,r建树,而权值线段树是以大小来建树,树上寸的信息是该权值的数量,而通过建树时二分从小到大的性质,可以用这个解决一些有序的东西

这个视频讲得很好,还有主席树的东西
https://www.bilibili.com/video/av16552942/?p=1&share_source=qq&share_medium=ipad&bbid=ZB4403B09F09305A44E7A4F9174F27CBF8AF&ts=1565180826
模板BZOJ3224
这道题主要是平衡树的模板,还未学习挖坑…

#include
#include
using namespace std;
#define N 100001
struct Data
{
    int v,p;
}t[N];
bool cmp(const Data &a,const Data &b)
{
    return a.v<b.v;
}
int T[N<<2],op[N],a[N],n,ma[N],e;
void Update(int p,int v,int rt,int l,int r)//插入删除
{
    T[rt]+=v;
    if(l==r) return;
    int m=(l+r>>1);
    if(p<=m) Update(p,v,rt<<1,l,m);
    else Update(p,v,rt<<1|1,m+1,r);
}
int Kth(int K,int rt,int l,int r)//K小值  根据二分自带性质 左边权值一定比右边小
{
    if(l==r) return l;
    int m=(l+r>>1);
    if(T[rt<<1]>=K) return Kth(K,rt<<1,l,m);
    return Kth(K-T[rt<<1],rt<<1|1,m+1,r);
}
int Rank(int p,int rt,int l,int r)//排名
{
    if(r<p) return T[rt];
    int m=(l+r>>1),res=0;
    res+=Rank(p,rt<<1,l,m);
    if(m<p-1) res+=Rank(p,rt<<1|1,m+1,r);
    return res;
}
int Findp(int rt,int l,int r)
{
    if(l==r) return l;
    int m=(l+r>>1);
    if(T[rt<<1|1]) return Findp(rt<<1|1,m+1,r);
    return Findp(rt<<1,l,m);
}
int Pre(int p,int rt,int l,int r)
{
    if(r<p)
      {
          if(T[rt]) return Findp(rt,l,r);
          return 0;
      }
    int m=(l+r>>1),Re;
    if(m<p-1 && T[rt<<1|1] && (Re=Pre(p,rt<<1|1,m+1,r))) return Re;
    return Pre(p,rt<<1,l,m);
}
int Findn(int rt,int l,int r)
{
    if(l==r) return l;
    int m=(l+r>>1);
    if(T[rt<<1]) return Findn(rt<<1,l,m);
    return Findn(rt<<1|1,m+1,r);
}
int Nex(int p,int rt,int l,int r)
{
    if(p<l)
      {
          if(T[rt]) return Findn(rt,l,r);
          return 0;
      }
    int m=(l+r>>1),Re;
    if(p<m && T[rt<<1] && (Re=Nex(p,rt<<1,l,m))) return Re;
    return Nex(p,rt<<1|1,m+1,r);
}
int main()
{
//    freopen("bzoj3224.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
      {
          scanf("%d%d",&op[i],&t[i].v);
          t[i].p=i;
      }
    sort(t+1,t+1+n,cmp);
    ma[a[t[1].p]=++e]=t[1].v;
    for(int i=2;i<=n;i++)
      {
        if(t[i].v!=t[i-1].v) ++e;
        ma[a[t[i].p]=e]=t[i].v;
      }
    for(int i=1;i<=n;i++)
      {
        if(op[i]==1) Update(a[i],1,1,1,e);
        else if(op[i]==2) Update(a[i],-1,1,1,e);
        else if(op[i]==3) printf("%d\n",Rank(a[i],1,1,e)+1);
        else if(op[i]==4) printf("%d\n",ma[Kth(ma[a[i]],1,1,e)]);
        else if(op[i]==5) printf("%d\n",ma[Pre(a[i],1,1,e)]);
        else printf("%d\n",ma[Nex(a[i],1,1,e)]);
      }
    return 0;
}

你可能感兴趣的:(模板,线段树)