線段樹模板

單點更新,區間查詢

struct node{
 	int l,r,max;
}a[MAX_N*4];
int b[MAX_N];
void update(int k){
 	a[k].max=max(a[k<<1].max,a[k<<1|1].max);
}
void build(int k,int l,int r){
 	a[k].l=l;a[k].r=r;
 	if(l==r){
  		a[k].max=b[l];
  		return;
 	}
 	int mid=(l+r)>>1;
 	build(k<<1,l,mid);
 	build(k<<1|1,mid+1,r);
 	update(k);
}
void change(int k,int x,int y){
 	if(a[k].l==a[k].r){
  		a[k].max=y;
  		return;
 	}
 	int mid=(a[k].l+a[k].r)/2;
 	if(x<=mid)
 	change(k<<1,x,y);
 	else
 	change(k<<1|1,x,y);
 	update(k);
}
int query(int k,int l,int r){
 	if(a[k].l>=l&&a[k].r<=r)
 	return a[k].max;
 	int mid=(a[k].l+a[k].r)>>1;
 	int x=0;
 	if(r>mid)
 	x=max(x,query(k<<1|1,l,r));
 	if(l<=mid)
 	x=max(x,query(k<<1,l,r));
 	return x;
}

區間更新,單點查詢

struct node{
	int l,r,sum,lazy;
}a[MAX_N*4];
void update(int k){
 	a[k].sum=a[k<<1].sum+a[k<<1|1].sum;
}
void build(int k,int l,int r){
 	a[k].l=l;a[k].r=r;a[k].lazy=0;
 	if(l==r){
  		a[k].sum=1;
  		return;
 	}
 	int mid=(l+r)>>1;
 	build(k<<1,l,mid);
 	build(k<<1|1,mid+1,r);
 	update(k);
}
void pushdown(int k){
 	if(a[k].lazy==0)
 	return;
 	a[k<<1].lazy=a[k<<1|1].lazy=a[k].lazy;
 	a[k<<1].sum=(a[k<<1].r-a[k<<1].l+1)*a[k].lazy;
 	a[k<<1|1].sum=(a[k<<1|1].r-a[k<<1|1].l+1)*a[k].lazy;
 	a[k].lazy=0;
}
void changeS(int k,int l,int r,int y){
 	if(a[k].l>=l&&a[k].r<=r){
  		a[k].sum=(a[k].r-a[k].l+1)*y;
  		a[k].lazy=y;
  		return;
 	}
 	pushdown(k);
 	int mid=(a[k].l+a[k].r)>>1;
 	if(l<=mid)
 	changeS(k<<1,l,r,y);
 	if(r>mid)
 	changeS(k<<1|1,l,r,y);
 	update(k);
}
int find(int k,int x){
 	if(a[k].l==a[k].r){
  		return a[k].sum;
 	}
 	pushdown(k);
 	int mid=(a[k].l+a[k].r)>>1;
 	if(x<=mid)
 	return find(k<<1,x);
 	else
 	return find(k<<1|1,x);
}

區間更新,區間查詢

struct node{
 	int l,r,sum,lazy;
}a[MAX_N*4];
void update(int k){
 	a[k].sum=a[k<<1].sum+a[k<<1|1].sum;
}
void build(int k,int l,int r){
 	a[k].l=l;a[k].r=r;a[k].lazy=0;
 	if(l==r){
  		a[k].sum=1;
  		return;
 	}
 	int mid=(l+r)>>1;
 	build(k<<1,l,mid);
 	build(k<<1|1,mid+1,r);
 	update(k);
}
void pushdown(int k){
 	if(a[k].lazy==0)
 	return;
 	a[k<<1].lazy=a[k<<1|1].lazy=a[k].lazy;
 	a[k<<1].sum=(a[k<<1].r-a[k<<1].l+1)*a[k].lazy;
 	a[k<<1|1].sum=(a[k<<1|1].r-a[k<<1|1].l+1)*a[k].lazy;
 	a[k].lazy=0;
}
void changeS(int k,int l,int r,int y){
 	if(a[k].l>=l&&a[k].r<=r){
  		a[k].sum=(a[k].r-a[k].l+1)*y;
  		a[k].lazy=y;
  		return;
 	}
 	pushdown(k);
 	int mid=(a[k].l+a[k].r)>>1;
 	if(l<=mid)
 	changeS(k<<1,l,r,y);
 	if(r>mid)
 	changeS(k<<1|1,l,r,y);
 	update(k);
}
int query(int k,int l,int r){	
 	if(a[k].l>=l&&a[k].r<=r)
 	return a[k].sum;
 	pushdown(k);
 	int mid=(a[k].l+a[k].r)>>1;
 	int x=0;
 	if(l<=mid)
 	x+=query(k<<1,l,r);
 	if(r>mid)
 	x+=query(k<<1|1,l,r);
 	return x;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章