题目背景
这是个非常经典的主席树入门题——静态区间第K小
数据已经过加强,请使用主席树。同时请注意常数优化
题目描述
如题,给定N个正整数构成的序列,将对于指定的闭区间查询其区间内的第K小值。
输入输出格式
输入格式:
第一行包含两个正整数N、M,分别表示序列的长度和查询的个数。 第二行包含N个正整数,表示这个序列各项的数字。 接下来M行每行包含三个整数 l, r, kl,r,k , 表示查询区间[l, r][l,r]内的第k小值。输出格式:
输出包含k行,每行1个正整数,依次表示每一次查询的结果输入输出样例
输入样例#1:
5 5 25957 6405 15770 26287 26465 2 2 1 3 4 1 4 5 1 1 2 2 4 4 1输出样例#1:
6405 15770 26287 25957 26287样例数据说明:
N=5,数列长度为5,数列从第一项开始依次为
25957, 6405, 15770, 26287, 26465 第一次查询为[2, 2]区间内的第一小值,即为6405 第二次查询为[3, 4]区间内的第一小值,即为15770 第三次查询为[4, 5]区间内的第一小值,即为26287 第四次查询为[1, 2]区间内的第二小值,即为25957 第五次查询为[4, 4]区间内的第一小值,即为26287分析:
练板子tip
离散化注意
sort(a+1,a+1+n,cmp);for (int i=1;i<=n;i++){ num[a[i].id]=i; V[i]=a[i].x;}
在build的时候now和top的关系不要搞混
不要忘了元素个数的增加
top++;t[top]=t[now];now=top;t[now].sum++; //元素个数
询问
int ask(int x,int y,int l,int r,int k){ if (l==r) return l; //权值线段树,位置就是离散值 int tmp=t[t[y].l].sum-t[t[x].l].sum; int mid=(l+r)>>1; if (tmp>=k) return ask(t[x].l,t[y].l,l,mid,k); else return ask(t[x].r,t[y].r,mid+1,r,k-tmp);}
这里写代码片#include#include #include #include using namespace std;const int N=200001;struct node{ int l,r,sum;};node t[N*40];int top=0,n,root[N],num[N],V[N],m;struct nd{ int x,id;};nd a[N];int cmp(const nd &a,const nd &b){ return a.x >1; if (z<=mid) build(t[now].l,l,mid,z); else build(t[now].r,mid+1,r,z); }int ask(int x,int y,int l,int r,int k){ if (l==r) return l; //权值线段树,位置就是离散值 int tmp=t[t[y].l].sum-t[t[x].l].sum; int mid=(l+r)>>1; if (tmp>=k) return ask(t[x].l,t[y].l,l,mid,k); else return ask(t[x].r,t[y].r,mid+1,r,k-tmp);}int main(){ scanf("%d%d",&n,&m); for (int i=1;i<=n;i++) scanf("%d",&a[i].x),a[i].id=i; sort(a+1,a+1+n,cmp); for (int i=1;i<=n;i++) { num[a[i].id]=i; V[i]=a[i].x; } for (int i=1;i<=n;i++) { root[i]=root[i-1]; build(root[i],1,n,num[i]); } for (int i=1;i<=m;i++) { int x,y,k; scanf("%d%d%d",&x,&y,&k); printf("%d\n",V[ask(root[x-1],root[y],1,n,k)]); } return 0;}