POJ-2104(主席树)

示例题目: POJ2104 K-th Number

题目大意:静态区间第 k <script type="math/tex" id="MathJax-Element-1">k</script>大
解题思路:主席树

#include<iostream>
#include<cstdio>
#include<string>
#include<cmath>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const int INF=0x3f3f3f3f;
const int MAXN=1e5+5;
const int M=MAXN*30;
int n,q,m,tot;
int a[MAXN],t[MAXN];
int T[MAXN],lson[M],rson[M],c[M];

void inithash()
{
  for(int i=1;i<=n;i++)
  {
    t[i]=a[i];
  }
  sort(t+1,t+1+n);
  m=unique(t+1,t+1+n)-(t+1);
}

int build(int l,int r)
{
  int root=tot++;
  c[root]=0;
  if(l!=r)
  {
    int mid=(l+r)>>1;
    lson[root]=build(l,mid);
    rson[root]=build(mid+1,r);
  }
  return root;
}

int Hash(int x)
{
  return lower_bound(t+1,t+1+m,x)-t;
}

int update(int root,int pos,int val)
{
  int newroot=tot++,tmp=newroot;
  c[newroot]=c[root]+val;
  int l=1,r=m;
  while(l<r)
  {
    int mid=(l+r)>>1;
    if(pos<=mid)
    {
      lson[newroot]=tot++;
      rson[newroot]=rson[root];
      newroot=lson[newroot];
      root=lson[root];
      r=mid;
    }else
    {
        rson[newroot]=tot++;
        lson[newroot]=lson[root];
        newroot=rson[newroot];
        root=rson[root];
        l=mid+1;
    }
    c[newroot]=c[root]+val;
  }
  return tmp;
}

int query(int leftroot,int rightroot,int k)
{
  int l=1,r=m;
  while(l<r)
  {
    int mid=(l+r)>>1;
    if(c[lson[leftroot]]-c[lson[rightroot]]>=k)
    {
      r=mid;
      leftroot=lson[leftroot];
      rightroot=lson[rightroot];
    }else
    {
      l=mid+1;
      k-=(c[lson[leftroot]]-c[lson[rightroot]]);
      leftroot=rson[leftroot];
      rightroot=rson[rightroot];
    }
  }
  return l;
}

int main()
{
  while(scanf("%d%d", &n,&q)!=EOF)
  {
    tot=0;
    for(int i=1;i<=n;i++)
    {
      scanf("%d",&a[i] );
    }
    inithash();
    T[n+1]=build(1,m);
    for(int i=n;i;i--)
    {
      int pos=Hash(a[i]);
      T[i]=update(T[i+1],pos,1);
    }
    while(q--)
    {
      int l,r,k;
      scanf("%d%d%d",&l,&r,&k );
      printf("%d\n",t[query(T[l],T[r+1],k)]);
    }
  }
}
/*
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

5
6
3
*/
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值