码迷,mamicode.com
首页 > 其他好文 > 详细

P3834 【模板】主席树

时间:2019-10-27 12:50:38      阅读:62      评论:0      收藏:0      [点我收藏+]

标签:ima   http   blank   query   info   src   div   alt   printf   

技术图片

 技术图片

题目地址


注意点:

  • MAXN可开到1e7.

#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=1e7,INF=2e9;
struct Node{
	int ls,rs;
	int cnt;
}tr[MAXN];
int nodeCnt=0;
void insert(int p,int &q,int l,int r,int val){
	if(!q)q=++nodeCnt;
	if(l==r){
		tr[q].cnt=tr[p].cnt+1;
		return;
	}
	int mid=(l+r)>>1;
	if(val<=mid){
		tr[q].rs=tr[p].rs;
		insert(tr[p].ls,tr[q].ls,l,mid,val);
	}else{
		tr[q].ls=tr[p].ls;
		insert(tr[p].rs,tr[q].rs,mid+1,r,val);
	}
	tr[q].cnt=tr[tr[q].ls].cnt+tr[tr[q].rs].cnt;
}
int query(int p,int q,int l,int r,int k){
	if(l==r)return l;
	int lsum=tr[tr[q].ls].cnt-tr[tr[p].ls].cnt;
	int mid=(l+r)>>1;
	if(k<=lsum)return query(tr[p].ls,tr[q].ls,l,mid,k);
	else return query(tr[p].rs,tr[q].rs,mid+1,r,k-lsum);
}
int root[MAXN];
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		int tmp;
		scanf("%d",&tmp);
		insert(root[i-1],root[i],-INF,INF,tmp);
	}
	for(int i=1;i<=m;i++){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",query(root[l-1],root[r],-INF,INF,k));
	}
	return 0;
}

  

 

P3834 【模板】主席树

标签:ima   http   blank   query   info   src   div   alt   printf   

原文地址:https://www.cnblogs.com/zbsy-wwx/p/11746889.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!