求区间 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;
}