主席树 模板

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define inf 2147483647
#define mem0(a) memset(a,0,sizeof(a))
#define rep(i,a,b) for(int i=a;i>1;
	build(lc[t],left,mid);
	build(rc[t],mid+1,right);
}
int modify(int p_root,int left,int right) { //加入一个数就相当于加到第几个数的树 就改变从根到叶子(该点为第几小)路径上的sum都加一
	int root=++node_cnt;//同样的,给每一个结点赋值
	lc[root]=lc[p_root];
	rc[root]=rc[p_root];
	sum[root]=sum[p_root]+1;//直接先连接原来的左右子树 然后该区间的值加一
	//一直走到叶子节点的路径上的值都加一
	if(left==right)return root;
	int mid=(left+right)>>1;
	if(p<=mid)lc[root]=modify(lc[root],left,mid);//如果左子树涉及P,设立新结点,改变左子树的结点值
	else rc[root]=modify(rc[root],mid+1,right);//如果右子树涉及P,设立新结点,改变右子树的结点值
	return root;//记得返回该区间的结点值,该结点是父节点的左结点或右结点,确定结点值
}
int query(int u,int v,int left,int right,int k) {
	int ans,x=sum[lc[v]]-sum[lc[u]];//x表示,[left,right]区间,左子树(左区间)有几个数
	if(left==right)return left;
	int mid=(left+right)>>1;
	if(x>=k)ans=query(lc[u],lc[v],left,mid,k);//如果左边的数大于等于K个,说明第K小的数在左子树(或者左区间)
	else
		ans=query(rc[u],rc[v],mid+1,right,k-x);//反之,就在右区间的第k-x个小的数
	return ans;
}
int main() {
	int n,m;
// freopen("testdata.in","r",stdin);
// freopen("res.txt","w",stdout);
	int T;
	scanf("%d",&T);
	while(T--) {
        node_cnt=0;
        rt[0]=0;
		scanf("%d %d",&n,&m);
		for(int i=1; i<=n; i++) {
			scanf("%d",&a[i]);
			b[i]=a[i];
		}
		sort(b+1,b+1+n);
		int q=unique(b+1,b+1+n)-b-1;//q表示不重复的数有几个,也表示主席树的最大右区间,[1,q]
		//区间[x,y]的值表示大于等于x,小于等于y的数有几个,这个用sum[k](k,结点值,代表这个区间)数组来记录
		//那么,结合下面的p和上面的q来看,这段程序其实将N个元素离散成1~q,然后再插入1~q插入到主席树中

		build(rt[0],1,q);//从0开始建 初始状态就是有0个数
		for(int i=1; i<=n; i++) {
			p=lower_bound(b+1,b+1+q,a[i])-b;
//这里p得到的是什么呢,它是a[i]这个元素在序列中是第几大(也就是在b中的下标)的数
			rt[i]=modify(rt[i-1],1,q);//每插入第i个数就生成第i个状态的树 第i棵树的访问方式由该函数返回根节点
			//每次都是在前一个状态的树上建立新树
		}
		while(m--) {
			int l,r,k;
			scanf("%d %d %d",&l,&r,&k);
			int ans=query(rt[l-1],rt[r],1,q,k);//得到的ans表示在b数组里面第ans小的数
			printf("%d\n",b[ans]);//最后输出b[ans]就是答案
		}
	}
}


 

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