ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

【主席树】【模板】

2022-02-10 10:00:30  阅读:136  来源: 互联网

标签:ch return int mid 模板 ls 主席 getchar


【主席树】【模板】

刚学主席树,理解还不是很深,先粘一下板子吧。

模板1

Code

#include<bits/stdc++.h>
using namespace std;

inline int read()
{
	register int x=0,w=1;
	register char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if(ch=='-') {w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
	return x*w;
}
inline void write(int x)
{
	if(x<0){x=~(x-1);putchar('-');}
	if(x>9) write(x/10);
	putchar(x%10+'0'); 
}
const int N=1e6+10;
int rt[N];
struct node{
	int ls,rs,val;
}t[N*25];
int n,m,a[N];
int tot;
int build(int l,int r)
{
	int p=++tot;
	if(l==r){
		t[p].val=a[l];
		return p;
	}
	int mid=l+r>>1;
	t[p].ls=build(l,mid);
	t[p].rs=build(mid+1,r);
	return p;
}
int insert(int now,int l,int r,int pos,int val)
{
	int p=++tot;
	t[p]=t[now];
	if(l==r){
		t[p].val=val;
		return p;
	}
	int mid=l+r>>1;
	if(pos<=mid) t[p].ls=insert(t[now].ls,l,mid,pos,val);
	else t[p].rs=insert(t[now].rs,mid+1,r,pos,val);
	return p;
}
int query(int x,int l,int r,int pos)
{
	if(l==r) return t[x].val;
	int mid=l+r>>1;
	if(pos<=mid) return query(t[x].ls,l,mid,pos);
	return query(t[x].rs,mid+1,r,pos);
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i) a[i]=read();
    rt[0]=build(1,n);
    for(int i=1;i<=m;++i)
    {
    	int r=read(),o=read(),pos=read();
    	if(o==1){
    		int val=read();
    		rt[i]=insert(rt[r],1,n,pos,val);
		}
		else{
		write(query(rt[r],1,n,pos));puts(""); 
		rt[i]=rt[r];
		}
	}
	return 0;
}

模板2(静态区间第k小)

#include<bits/stdc++.h>
using namespace std;

inline int read()
{
	register int x=0,w=1;
	register char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
	if(ch=='-') {w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
	return x*w;
}
const int N=2e5+100;
int n,m,a[N],rk[N];
int rt[N],tot;
struct node{
	int ls,rs,sum;
}t[N*40];
int build(int l,int r)
{
	int p=++tot;
	if(l==r) return p;
	int mid=l+r>>1;
	t[p].ls=build(l,mid);
	t[p].rs=build(mid+1,r);
	return p;
}
void pushup(int x)
{
	t[x].sum=t[t[x].ls].sum+t[t[x].rs].sum;
}
int insert(int x,int l,int r,int pos)
{
	int p=++tot;
	t[p]=t[x];
	if(l==r){
		t[p].sum++;
		return p;
	}
	int mid=l+r>>1;
	if(pos<=mid) t[p].ls=insert(t[p].ls,l,mid,pos);
	else t[p].rs=insert(t[p].rs,mid+1,r,pos);
	pushup(p);
	return p;
}
inline void write(int x)
{
	if(x<0) {putchar('-');x=~(x-1);}
	if(x>9) write(x/10);
	putchar(x%10+'0');
}
int query(int p,int q,int l,int r,int k)
{
	if(l==r) return rk[l];
	int mid=l+r>>1;
	int lsum=t[t[q].ls].sum-t[t[p].ls].sum;
	if(k<=lsum) return query(t[p].ls,t[q].ls,l,mid,k);
	return query(t[p].rs,t[q].rs,mid+1,r,k-lsum);
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i){
    	a[i]=rk[i]=read();
	}
	sort(rk+1,rk+n+1);
	int cnt=unique(rk+1,rk+n+1)-rk-1;
	for(int i=1;i<=n;++i){
		a[i]=lower_bound(rk+1,rk+cnt+1,a[i])-rk;
	}
	rt[0]=build(1,cnt);
	for(int i=1;i<=n;++i){
		rt[i]=insert(rt[i-1],1,cnt,a[i]);
	}
	for(int i=1;i<=m;++i)
	{
		int l=read(),r=read(),k=read();
		write(query(rt[l-1],rt[r],1,cnt,k));puts("");
	}
	return 0;
}

标签:ch,return,int,mid,模板,ls,主席,getchar
来源: https://www.cnblogs.com/glq-Blog/p/15877910.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有