主席树真好用,收回之前的话
一个长度为N的整数序列,编号0 - N - 1。进行Q次查询,查询编号i至j的所有数中,第K大的数是多少。
例如: 1 7 6 3 1。i = 1, j = 3,k = 2,对应的数为7 6 3,第2大的数为6。
输入
第1行:1个数N,表示序列的长度。(2 <= N <= 50000)
第2 - N + 1行:每行1个数,对应序列中的元素。(0 <= S[i] <= 10^9)
第N + 2行:1个数Q,表示查询的数量。(2 <= Q <= 50000)
第N + 3 - N + Q + 2行:每行3个数,对应查询的起始编号i和结束编号j,以及k。(0 <= i <= j <= N - 1,1 <= k <= j - i + 1)
输出
共Q行,对应每一个查询区间中第K大的数。
输入样例
5
1
7
6
3
1
3
0 1 1
1 3 2
3 4 2
输出样例
7
6
1
分析:
这类问题比较有针对性,目前看来遇到的有关线段树中关于第k级大或者小的逻辑关系问题,都可以用主席树来解决,前有一段时间认为主席树在已有线段树复杂度的基础上进行合并会变得更加复杂,但实际上是在线段树离散化的基础上进行的单线段树操作,会用到树状数组的思想,
可以理解为线段树plus
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5+10;
int num[MAXN],root[MAXN];
struct node
{
int l,r,sum;
} T[MAXN*40];
int cnt,n,m;
vector<int> vec;
int getid(int x)
{
return lower_bound(vec.begin(),vec.end(),x) - vec.begin() + 1;
}
void init()
{
vec.clear();
cnt = 0;
root[0] = 0;
T[0].l = T[0].r = T[0].sum = 0;
}
void update(int l, int r, int &x, int y, int pos)
{
T[++cnt] = T[y];
T[cnt].sum++;
x = cnt;
if(l == r) return;
int mid = (l+r) >> 1;
if(pos <= mid) update(l,mid,T[x].l,T[y].l,pos);
else update(mid+1,r,T[x].r,T[y].r,pos);
}
int query(int l, int r, int x, int y, int k)
{
if(l == r) return l;
int mid = (l+r) >> 1;
int sum = T[T[y].r].sum - T[T[x].r].sum;
if(sum >= k) return query(mid+1,r,T[x].r,T[y].r,k);
else return query(l,mid,T[x].l,T[y].l,k-sum);
}