首页 > 代码库 > HDU 2665 Kth number
HDU 2665 Kth number
题解:求区间K小,函数式线段树模板题。
#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N=3000005;struct node{int num,id;}a[N];int T,n,m,x,y,z,tot,b[N],head[N],son[N][2],data[N];int cmp(node a,node b){return a.num<b.num;}int make(int l,int r){ int t=++tot,mid=(l+r)>>1; if(l==r)return t; son[t][0]=make(l,mid);son[t][0]=make(mid+1,r); return t;}int insert(int x,int pos,int l,int r){ int t=++tot,mid=(l+r)>>1; data[t]=data[pos]+1; if(l==r)return t; if(x<=mid)son[t][1]=son[pos][1],son[t][0]=insert(x,son[pos][0],l,mid); else{son[t][0]=son[pos][0];son[t][1]=insert(x,son[pos][1],mid+1,r);} return t;} int ask(int x,int y,int k,int l,int r){ int mid=(l+r)>>1; if(l==r)return l; int rk=data[son[y][0]]-data[son[x][0]]; if(k<=rk)return ask(son[x][0],son[y][0],k,l,mid); else return ask(son[x][1],son[y][1],k-rk,mid+1,r);}int main(){ scanf("%d",&T); while(T--){ tot=0; scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){scanf("%d",&a[i].num);a[i].id=i;} sort(a+1,a+n+1,cmp); for(int i=1;i<=n;i++)b[a[i].id]=i; head[0]=make(1,n); for(int i=1;i<=n;i++)head[i]=insert(b[i],head[i-1],1,n); for(int i=1;i<=m;i++){ scanf("%d%d%d",&x,&y,&z); if(x>y)swap(x,y); printf("%d\n",a[ask(head[x-1],head[y],z,1,n)].num); } }return 0;}
HDU 2665 Kth number
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。