1175 区间中第K大的数

主席树真好用,收回之前的话
一个长度为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);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值