【Luogu3834】【模板】可持久化线段树 2(主席树)

[ L i n k \frak{Link} Link]


求区间 kth 显然可以离散化权值,对每个位置用树维护此前每个权值的出现次数,
于是树上可以查询得到一定范围内所有权值的出现次数和

主席树除了第一个版本之外所有版本相当于 copy 了上一版本的树只修改新的结点
然后为了加速,就变成除了新的数据之外直接连接到上一版本的结点

查询的时候在 L-1 和 R 两个版本上同时跑,实时导航区间 kth
O ( m log ⁡ n ) O(m\log n) O(mlogn)
坑只有一个,我标上了


#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
const int MAXN = 2e5 + 5;
int n, m, a[MAXN], valori[MAXN];
pair<int, int> b[MAXN];

class SegUnit
{
public:
	int Lc, Rc, Vl;
	
//	SegUnit(): Id(0) { }
	SegUnit(int Ar2=0, int Ar3=0, int Ar4=0)
	{
		Lc = Ar2;
		Rc = Ar3;
		Vl = Ar4;
	}
};

int SegTot = 0, rt[MAXN]={};

template <class T>
class SegmentTree
{
public:
	SegUnit Seg[MAXN<<5];
	
	int Query(int PosL, int PosR, int L, int R, int k)
	{
		if (L >= R) return L;
		if (L < R)
		{
			int Mid = L + R >> 1;
			if (Seg[Seg[PosR].Lc].Vl - Seg[Seg[PosL].Lc].Vl >= k) return Query(Seg[PosL].Lc, Seg[PosR].Lc, L, Mid, k);
			else return Query(Seg[PosL].Rc, Seg[PosR].Rc, Mid + 1, R, k - Seg[Seg[PosR].Lc].Vl + Seg[Seg[PosL].Lc].Vl); //*!!!!!
		}
	}
	
	int Init(int L, int R)
	{
		int id = ++SegTot;
		Seg[id] = SegUnit(0, 0, 0);
		if (L<R) 
		{
			int Mid = L + R >> 1;
			Seg[id].Lc=Init(L, Mid);
			Seg[id].Rc=Init(Mid+1, R);
		}
		return id;
	}
	
	int Update(int Pre, int L, int R, int val)
	{
		int id = ++SegTot;
		Seg[id] = SegUnit(Seg[Pre].Lc, Seg[Pre].Rc, Seg[Pre].Vl+1);
		if (L < R)
		{
			int Mid = L + R >> 1;
			if (val <= Mid) Seg[id].Lc = Update(Seg[Pre].Lc, L, Mid, val);
			else Seg[id].Rc = Update(Seg[Pre].Rc, Mid + 1, R, val);
		}
		return id;
	}
	
};
SegmentTree<int> MyTree;

bool cmp(pair<int, int>a, pair<int, int>b)
{
	return a.first < b.first;
}

int tot = 0;
void Discretize()
{
	for (register int i = 1; i <= n; ++i)
	{
		b[i] = make_pair(a[i], i);
	}
	sort(b+1, b+1+n, cmp);
	for (register int i = 1; i <= n; ++i)
	{
		if (i == 1 || b[i].first != b[i-1].first) valori[++tot]=b[i].first;
		a[b[i].second]=tot;
	}
}

int main()
{	
	scanf("%d%d", &n, &m);
	for (register int i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
	}
	Discretize();
	MyTree.Init(0, tot+1);
	rt[0] = 1;
	for (register int i = 1; i <= n; ++i)
	{
		rt[i] = MyTree.Update(rt[i-1], 1, tot, a[i]);
	}
	for (register int l, r, k, i = 1; i <= m; ++i)
	{
		scanf("%d%d%d", &l, &r, &k);
		printf("%d\n", valori[MyTree.Query(rt[l-1], rt[r], 1, tot, k)]);
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值