51nod 1175 区间中第K大的数

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。
Input
第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)
Output
共Q行,对应每一个查询区间中第K大的数。
Input示例
5
1
7
6
3
1
3
0 1 1
1 3 2
3 4 2
Output示例
7
6
1
相关问题

#include
#include
using namespace std;
const int   
  N=500005,  
  logN=20;  
int n,Tcnt;  
int root[N];  
struct ChairTree{  
  int l,r,num;  
}ct[N*logN];  
int read(){  
  int x=0,f=1;char ch=getchar();  
  while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}  
  while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}  
  return x*f;  
}  
void insert(int L,int R,int &x,int val){  
  ct[Tcnt++]=ct[x];  
  x=Tcnt-1;  
  ct[x].num++;  
  if (L==R) return;  
  int mid=(L+R)>>1;  
  if (val<=mid) insert(L,mid,ct[x].l,val);  
  else insert(mid+1,R,ct[x].r,val);  
}  
int query(int L,int R,int ll,int rr,int k){  
  if (L==R) return L;  
  int mid=(L+R)>>1; 
  int t=ct[ct[rr].l].num-ct[ct[ll].l].num;  
  if (t>=k) query(L,mid,ct[ll].l,ct[rr].l,k);
  else query(mid+1,R,ct[ll].r,ct[rr].r,k-t);
}  
int main(){
  n=read();
  root[0]=0,Tcnt=1;
  for (int i=1;i<=n;++i){
    root[i]=root[i-1];
    insert(1,1e9,root[i],read());
    }
    int Q=read();
    while (Q--){
      int i=read(),j=read(),k=read();
      ++i;++j;
      printf("%d\n",query(1,1e9,root[i-1],root[j],j-i+1-k+1));
    }
    return 0;
}

你可能感兴趣的:(主席树)