洛谷P3834主席树模板题(+离散化)

题目链接 P3834 【模板】可持久化线段树 2
一般主席树应用场景，求区间第k大数字，主席树维护了一个权值线段树的前缀和。空间复杂度一般为n<<5。因为一个普通的线段树是2 n - 1个节点，假设修改m次，每次修改最多增加一条链也就是logV个节点，那么整体的空间复杂度为2n - 1 + nlogV，是nlogV级别。其中V是权值线段树的最大权值，所以离散化之后能降低一些树的层数，但是也可以不离散化。
离散化的代码

#include<bits/stdc++.h>
using namespace std;
#define LL long long 
#define maxn 200005
#define maxm 1000005
#define INF 0x3f3f3f3f
int n,m,a[maxn];
namespace disc {
	vector<LL>dv;
	int len = 0;
	template<typename T>
	int lenth(T&& f, int n){
		dv.clear();
		for(int i=1;i<=n;i++)dv.push_back(1ll * f[i]);
		sort(dv.begin(), dv.end());
        len=unique(dv.begin(), dv.end()) - dv.begin();
        return len;
	}
	int getid(int x){ return lower_bound( dv.begin(), dv.begin() + len, x) - dv.begin() + 1; }  // 最小从1开始的
	LL getval(int id){ return dv[id - 1]; }  // 查询最小也是从1开始的
}
namespace ct {
#define mid (l + r)/2
	int tot, rt[maxn],ls[maxn<<5],rs[maxn<<5],w[maxn<<5];
	void insert(int l, int r, int x, int &y, int v){
		y = ++tot;
		w[y] = w[x] + 1;
		if (l == r) return;
		ls[y]=ls[x];rs[y]=rs[x];
		if (v <= mid)insert(l, mid, ls[x], ls[y], v);
		else insert(mid + 1, r, rs[x], rs[y], v);
	}
	int query(int l, int r, int x, int y, int k){
		if (l == r) return l;
		int tmp = w[ls[y]] - w[ls[x]];
		if (k <= tmp)return query(l, mid, ls[x], ls[y], k);
		else return query(mid + 1, r, rs[x], rs[y], k - tmp);
	}
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int len = disc::lenth(a, n);
    for(int i=1;i<=n;i++)ct::insert(1, len, ct::rt[i-1], ct::rt[i], disc::getid(a[i]));
    for(int i=1,x,y,z;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        x--;
        int ans= disc::getval(ct::query(1, len, ct::rt[x], ct::rt[y], z));
        printf("%d\n",ans);
    }
    return 0;
}

以前写的不离散化的代码

#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
#define maxm 1000005
#define INF 0x3f3f3f3f
#define ll long long int
#define mod 1000000007
int n,m;
int a[maxn];
int rt[maxn],ls[maxn<<5],rs[maxn<<5],w[maxn<<5],v[maxn<<5],tot;
void insert(int l,int r,int x,int &y,int s)
{
    y=++tot;
    w[y]=w[x]+1;
    if(l==r)return ;
    ls[y]=ls[x];rs[y]=rs[x];
    int mid=(l+r)>>1;
    if(s<=mid)insert(l,mid,ls[x],ls[y],s);
    else insert(mid+1,r,rs[x],rs[y],s);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r)return l;
    int mid=(l+r)>>1;
    int tmp=w[ls[y]]-w[ls[x]];
    if(k<=tmp)return query(l,mid,ls[x],ls[y],k);
    else return query(mid+1,r,rs[x],rs[y],k-tmp);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)insert(-mod,mod,rt[i-1],rt[i],a[i]);
    for(int i=1,x,y,z;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        x--;
        int ans=query(-mod,mod,rt[x],rt[y],z);
        printf("%d\n",ans);
    }
    return 0;
}