初学主席树

原创 2016年08月29日 12:56:11

关于知识点的请看链接(我翻了n个blog,就这个写的最明白):点击打开链接

代码:(区间第k大)

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#define lson l, m
#define rson m+1, r
using namespace std;
const int N=1e5+5;
int L[N<<5], R[N<<5], sum[N<<5];
int tot;
int a[N], T[N], Hash[N];
int build(int l,int r)
{
       int rt=(++tot);
       sum[rt]=0;
       if(l<r)
       {

           int m=(r+l)>>1;
           L[rt]=build(lson);
           R[rt]=build(rson);
       }
    return rt;
}
int update(int pre,int l,int r,int x)
{
    int rt=(++tot);
    L[rt]=L[pre],R[rt]=R[pre],sum[rt]=sum[pre]+1;
    if(l<r)
    {
        int m=(r+l)>>1;
        if(x<=m)
        {
            L[rt]=update(L[pre],lson,x);
        }
        else
            R[rt]=update(R[pre],rson,x);
    }
    return rt;
}
int query(int u,int v,int l,int r,int k)
{
    if(l>=r)
        return l;
    int m=(l+r)>>1;
    int num=sum[L[v]]-sum[L[u]];
    if(num>=k)
    {
        return query(L[u],L[v],lson,k);
    }
    else
        return query(R[u],R[v],rson,k-num);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        tot=0;
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            Hash[i]=a[i];
        }
        sort(Hash+1,Hash+1+n);
        int d=unique(Hash+1,Hash+1+n)-Hash-1;
        T[0]=build(1,d);
        for(int i=1;i<=n;i++)
        {
            int x=lower_bound(Hash+1,Hash+d+1,a[i])-Hash;
            T[i]=update(T[i-1],1,d,x);
        }
        while(m--)
        {
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            int x=query(T[l-1],T[r],1,d,k);
            printf("%d\n",Hash[x]);
        }
    }
}

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<math.h>
#include<string.h>
#define ll long long
using namespace std;
const int N=1e6+6;
vector<int>p;
int cnt,n,m,x,y,k,root[N],a[N];
struct node{int l,r,sum;}T[N*40];
int getid(int x){ return lower_bound(p.begin(),p.end(),x)-p.begin()+1;}
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=(r+l)>>1;
    if(mid>=pos)
        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=(r+l)>>1;
    int sum=T[T[y].l].sum-T[T[x].l].sum;
    if(sum>=k)
        return query(l,mid,T[x].l,T[y].l,k);
    else
        return query(mid+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),p.push_back(a[i]);
    sort(p.begin(),p.end()),p.erase(unique(p.begin(),p.end()),p.end());
    for(int i=1;i<=n;i++) update(1,n,root[i],root[i-1],getid(a[i]));
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",p[query(1,n,root[x-1],root[y],k)-1]);
    }
    return 0;
}


版权声明:加油 加油 加油^-^

相关文章推荐

主席树.ppt

  • 2016年07月08日 09:18
  • 232KB
  • 下载

区间第k大 主席树代码

  • 2014年11月03日 16:06
  • 3KB
  • 下载

主席树 初学

现在才开始学主席树(←弱(。・・)ノ)。不过不带修改的话代码还是很简单的嘛。。。或者说应该叫可持久化线段树?     首先对数的区间进行离散化,这样下面的a[i]都默认为离散化以后的结果了。     ...

【主席树】51Nod 1175 区间中第K大的数

题面在这里主席树裸题……联赛快来了还是练练手速吧示例程序
  • linkfqy
  • linkfqy
  • 2017年10月30日 16:46
  • 497

[BZOJ4556][Tjoi2016&Heoi2016]字符串(后缀数组+二分+st表+主席树)

题目描述传送门题解思路清晰就很好写… 首先二分答案mid 找到Suffix(c),在height数组中向左向右分别二分最远的lcp为mid的后缀 用st表实现O(1)O(1)查询 然后就判断[...

主席树-区间第k大值(不带修改)

题目:poj2104 && hdu2665 题意:求区间第K大的值。 分析:资料1  主席树包含n棵线段树,这n棵线段树的形状完全相同。而且树与树之间有很大的重叠。 线段树root[i]表示数...
  • w20810
  • w20810
  • 2016年01月05日 12:55
  • 779

ZOJ 2112 Dynamic Rankings(主席树-动态第k大)

Description 给出一个长度为n的序列a,两种操作 C x v:将第x个元素的值改成v Q l r k:查询区间[l,r]中第k大的元素 Input 第一行为一个整数t表示用例组数,...
  • V5ZSQ
  • V5ZSQ
  • 2015年12月06日 16:09
  • 788

【主席树】POJ 2014 K-th Number

POJ 2014 K-th NumberDescriptionYou are working for Macrohard company in data structures department. ...

【主席树】Codechef Prefix XOR

题面在这里一道很巧妙的题目……构造前缀异或和 那么SjxorSi−1≥Sj−1xorSi−1S_jxorS_{i-1}\ge S_{j-1}xorS_{i-1} 考虑SjS_j和Sj−1S_{j-...
  • linkfqy
  • linkfqy
  • 2017年08月01日 12:30
  • 604

[BZOJ3772]精神污染(主席树+链剖)

ATP放学了来不及写摘要了。。。
  • FromATP
  • FromATP
  • 2017年01月11日 22:00
  • 286
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:初学主席树
举报原因:
原因补充:

(最多只允许输入30个字)