【POJ】2104 K-th Number 主席树

13 篇文章 0 订阅
6 篇文章 0 订阅

题目传送门
主席树什么真的好迷啊……
写一下自己对主席树的理解吧,好像就是一个线段树的前缀和。
首先我们先建出一棵空树,作为之后线段树的基础。
对每一个节点都建一棵线段树,但是也不是完全建出整棵线段树,而是对于不需要修改的节点就直接调用当前节点的对应节点作为儿子,对于需要修改的节点就建出新的节点作为儿子,并对新建出来节点进行修改。
然后发现这样的数据结构就可以有求区间第k大等骚操作了。
博主对于主席树还不是很熟练,现在还只会没有修改的求区间第k大,如果之后学到了更多的骚操作再来填坑吧。
回到这题,就是一个求区间第k大的裸题,直接主席树上就行了。
p.s.主席树要用到动态开点的知识,需要用到的节点为2*n+m*log2(n),“但是最好把内存开的大一些,防止爆炸。”——来自ZZK的教诲
附上AC代码:

#include <cstdio>
#include <cctype>
#include <algorithm>
#define mid (l+r>>1)
using namespace std;

const int N=1e5+10;
struct tree{
    int lt,rt,sum;
}t[2000010];
int n,m,a[N],b[N],root[N],size,x,y,w;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    a*=f;return;
}

inline void ist(int pre,int &k,int l,int r,int w){
    t[k=++size]=t[pre],++t[k].sum;
    if (l==r) return;
    if (mid>=w) ist(t[pre].lt,t[k].lt,l,mid,w);
    else ist(t[pre].rt,t[k].rt,mid+1,r,w);
    return;
}

inline int query(int pre,int k,int l,int r,int w){
    if (l==r) return b[l];
    int c=t[t[k].lt].sum-t[t[pre].lt].sum;
    if (c>=w) return query(t[pre].lt,t[k].lt,l,mid,w);
    else return query(t[pre].rt,t[k].rt,mid+1,r,w-c);
}

int main(void){
    read(n),read(m);
    for (int i=1; i<=n; ++i) read(a[i]),b[i]=a[i];
    sort(b+1,b+1+n);
    for (int i=1; i<=n; ++i) ist(root[i-1],root[i],1,n,lower_bound(b+1,b+1+n,a[i])-b);
    while (m--) read(x),read(y),read(w),printf("%d\n",query(root[x-1],root[y],1,n,w));
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值