我真弱。。。连主席树都不会。。。
主席树相当于多个线段树,由于相邻两棵线段树的节点的值只有少许不同,因此可以对于和前一棵树一样的子树一个指针指过去,无需操作,这样每棵树O(logn),总复杂度O(nlogn)。
以下是区间K大代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#define N 100005
#define M 4000000
using namespace std;
int n,m,a[N],b[N],c[N],cnt,root[N],L[M],R[M],v[M],len,x,y,k;
bool cmp(int x,int y){return a[x]<a[y];}
void add(int l,int r,int x,int &y,int k)
{
y=++len;v[y]=v[x]+1;
if (l==r) return;
int mid=(l+r)>>1;
if (k<=mid) R[y]=R[x],add(l,mid,L[x],L[y],k);
else L[y]=L[x],add(mid+1,r,R[x],R[y],k);
}
int qry(int x,int y,int k)
{
int l=1,r=cnt;
x=root[x-1];y=root[y];
while (l<r)
{
int mid=(l+r)>>1,t=v[L[y]]-v[L[x]];
if (t>=k) {r=mid;x=L[x];y=L[y];}
else {l=mid+1;x=R[x];y=R[y];k-=t;}
}
return l;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",&a[i]),b[i]=i;
sort(b+1,b+n+1,cmp);
for (int i=1;i<=n;i++)
{
if (i==1||a[b[i]]!=a[b[i-1]]) cnt++;
c[b[i]]=cnt;a[cnt]=a[i];
}
for (int i=1;i<=n;i++)
add(1,cnt,root[i-1],root[i],c[i]);
for (int i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&k);
printf("%d\n",a[b[qry(x,y,k)]]);
}
}