HUEL_ACM第二学期第四周：线段树

最新推荐文章于 2022-05-06 12:47:06 发布

重生之我是研究生

最新推荐文章于 2022-05-06 12:47:06 发布

阅读量327

点赞数 1

分类专栏： XCPC&CF&蓝桥杯_题解

本文链接：https://blog.csdn.net/Alanrookie/article/details/114673866

版权

XCPC&CF&蓝桥杯_题解专栏收录该内容

192 篇文章 5 订阅

订阅专栏

A模板题1

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=2e5+6;
int n,m;
struct node{
	int l,r;
	int v;
}tr[N*4];

void push_up(int u){
	tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}

void build(int u,int l,int r){
	tr[u]={l,r};
	if(l==r)return;
	int mid=l+r>>1;
	build(u<<1,l,mid),build(u<<1|1,mid+1,r);
}

int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].v;
	int mid=tr[u].l+tr[u].r>>1;
	if(r<=mid)return query(u<<1,l,r);
	else if(l>mid)return query(u<<1|1,l,r);
	else return max(query(u<<1,l,r),query(u<<1|1,l,r));
}

void modify(int u,int x,int v){
	if(tr[u].l==x&&tr[u].r==x)tr[u].v=v;
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(x<=mid)modify(u<<1,x,v);
		else modify(u<<1|1,x,v);
		push_up(u);
	}
}
int main(){
	while(~scanf("%d%d",&n,&m)){
	
	build(1,1,n);
	 
	for(int i=1;i<=n;i++){
		int x;
	    scanf("%d",&x);
		modify(1,i,x);
	}
	
	//getchar();
	while(m--){
		char s[10];
		int a,b;
		scanf("%s%d%d",s,&a,&b);
		//cout<<s<<" "<<a<<" "<<b<<endl;
		if(s[0]=='Q'){
			printf("%d\n",query(1,a,b));
		}
		else {
			modify(1,a,b);
		}
	}
}
}

B模板题2

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define int long long
using namespace std;
const int N=2e5+6;
int n,m,w[N];
struct node{
	int l,r;
	int sum;
	int add;
}tr[N*4];

void push_up(int u){
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}

void push_down(int u){
	node &fa=tr[u],&ls=tr[u<<1],&rs=tr[u<<1|1];
	if(fa.add){
		ls.add+=fa.add,ls.sum+=(ls.r-ls.l+1)*fa.add;
		rs.add+=fa.add,rs.sum+=(rs.r-rs.l+1)*fa.add;
		fa.add=0;     
	}
} 

void build(int u,int l,int r){
	if(l==r) tr[u]={l,r,w[l],0};
	else{
		tr[u]={l,r};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		push_up(u); 
	}
}

void modify(int u,int l,int r,int d){
	if(tr[u].l>=l&&tr[u].r<=r){
		tr[u].sum+=(tr[u].r-tr[u].l+1)*d;
		tr[u].add+=d; 
	}
	else{
		push_down(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid)modify(u<<1,l,r,d);
		else if(l>mid)modify(u<<1|1,l,r,d);
		else modify(u<<1,l,r,d),modify(u<<1|1,l,r,d);
		push_up(u);
	}
}

int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
	else{
		push_down(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid) return query(u<<1,l,r);
		else if(l>mid) return query(u<<1|1,l,r);
		else return query(u<<1,l,r)+query(u<<1|1,l,r);
		push_up(u);
	} 
}

main(){
	scanf("%lld%lld",&n,&m);
	
	for(int i=1;i<=n;i++)scanf("%lld",&w[i]);
	
	build(1,1,n);
	
	while(m--){
		char op[10];
		int l,r,d;
		scanf("%s%lld%lld",op,&l,&r);
		if(op[0]=='Q')printf("%lld\n",query(1,l,r));
		else{
			scanf("%lld",&d);
			modify(1,l,r,d);
		}
	}
}

C模板题3

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=2e5+6;
int n,m;
struct node{
	int l,r;
	int sum;
	int v;
}tr[N*4];

void push_up(int u){
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;   
}

void push_down(int u){
	node &fa=tr[u],&ls=tr[u<<1],&rs=tr[u<<1|1];
	if(fa.v){ 
	ls.v=fa.v,ls.sum=(ls.r-ls.l+1)*fa.v;
	rs.v=fa.v,rs.sum=(rs.r-rs.l+1)*fa.v;
	}
	fa.v=0; 
	    
}

void build(int u,int l,int r){
	if(l==r)tr[u]={l,r,1,1};
	else{
		push_down(u); 
		tr[u]={l,r};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		push_up(u); 
	}
}

void modify(int u,int l,int r,int x){
	if(tr[u].l>=l&&tr[u].r<=r){
		tr[u].v=x;
		tr[u].sum=(tr[u].r-tr[u].l+1)*x; 
		//push_down(u);
	}
	else{
		push_down(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid)modify(u<<1,l,r,x);
		else if(l>mid)modify(u<<1|1,l,r,x);
		else modify(u<<1,l,r,x),modify(u<<1|1,l,r,x);
		push_up(u);
	}
}

int query(int u,int l,int r){
	push_down(u);
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
	//push_down(u);
	int mid=tr[u].l+tr[u].r>>1;
	if(r<=mid)return query(u<<1,l,r);
	else if(l>mid)return query(u<<1|1,l,r);
	else return query(u<<1,l,r)+query(u<<1|1,l,r); 
}

int main(){
	int T;
	scanf("%d",&T);
	for(int TT=1;TT<=T;TT++){
		scanf("%d%d",&n,&m);
		
		build(1,1,n);
		
		while(m--){
			
			//cout<<"*****"<<query(1,1,n)<<endl;
			
			int l,r,x;
			scanf("%d%d%d",&l,&r,&x);
			modify(1,l,r,x);
			
			//cout<<"&&&&&"<<query(1,1,n)<<endl;
		}
		
		printf("Case %d: The total value of the hook is %d.\n",TT,query(1,1,n));
	}
}

D应用1：求最长连续区间，对于一个数组上的某个点来说，我们所求的包含这个点的最长连续区间其实是左边最近的断点到右边最近的断点之间的长度。

#include <iostream>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std;
const int N=1e5+6;
int n,m;
struct node{
	int l,r;
	int lpos,rpos;//分别记录本段区间内最左端的断点和最右端的断点位置 
}tr[N*4];

void push_up(int u){
	tr[u].lpos=max(tr[u<<1].lpos,tr[u<<1|1].lpos);
	tr[u].rpos=min(tr[u<<1].rpos,tr[u<<1|1].rpos);
} 

void build(int u,int l,int r){
	if(l==r)tr[u]={l,r,0,n+1};
	else{ 
		tr[u]={l,r};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		push_up(u);
	}
}

void modify(int u,int x,int lpos,int rpos){
	if(tr[u].l==x&&tr[u].r==x){
		tr[u].lpos=lpos;
		tr[u].rpos=rpos; 
	}
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(x<=mid)modify(u<<1,x,lpos,rpos);
		else modify(u<<1|1,x,lpos,rpos);
		push_up(u);
	}
}

int query1(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].lpos;
	int mid=tr[u].l+tr[u].r>>1;
	if(r<=mid)return query1(u<<1,l,r);
	else if(l>mid)return query1(u<<1|1,l,r);
	else return max(query1(u<<1,l,r),query1(u<<1|1,l,r));
}

int query2(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].rpos;
	int mid=tr[u].l+tr[u].r>>1;
	if(r<=mid)return query2(u<<1,l,r);
	else if(l>mid)return query2(u<<1|1,l,r);
	else return min(query2(u<<1,l,r),query2(u<<1|1,l,r));
}

int main(){
	while(~scanf("%d%d",&n,&m)){
		stack<int>q;
		build(1,1,n);
		
		while(m--){
			char s[10];
			scanf("%s",s);
			if(s[0]=='R'){
				int u=q.top();
				q.pop();
				modify(1,u,0,n+1); 
			}
			else{
				int x;
				scanf("%d",&x);
				if(s[0]=='D'){
					modify(1,x,x,x);
					q.push(x);
				}
				else {
					int lpos=query1(1,1,x);
					int rpos=query2(1,x,n);
					//cout<<lpos<<"****"<<rpos<<endl;
					printf("%d\n",max(0,rpos-lpos-1));
				}
			}
		}
	}
}

E:预处理DFS序后直接套线段树模板即可

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=2e5+6;
int n,m;
int h[N],e[N],ne[N],idx;
int l[N],r[N],timestamp;//分别记录以当前点为根节点时整棵树在序列中的左端点和右端点 
int w[N],cnt;
int din[N];
struct node{
	int l,r;
	int v;
}tr[N*4];


void add(int a,int b){
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs(int u){
	l[u]=++timestamp;
	for(int i=h[u];~i;i=ne[i]){
		int v=e[i];
		dfs(v);
	}
	r[u]=timestamp;
}

void push_down(int u){
	node &fa=tr[u],&ls=tr[u<<1],&rs=tr[u<<1|1];
	if(fa.v>0){
		ls.v=fa.v;
		rs.v=fa.v;
		fa.v=-1;     
	}
}

void build(int u,int l,int r){
	if(l==r)tr[u]={l,r,-1};
	else{
		tr[u]={l,r,-1};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
	}
}

void modify(int u,int l,int r,int x){
	if(tr[u].l>=l&&tr[u].r<=r)tr[u].v=x;
	else{
		push_down(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid)modify(u<<1,l,r,x);
		else if(l>mid)modify(u<<1|1,l,r,x);
		else modify(u<<1,l,r,x),modify(u<<1|1,l,r,x);
	}
}

int query(int u,int x){
	if(tr[u].l==x&&tr[u].r==x)return tr[u].v;
	push_down(u);
	int mid=tr[u].l+tr[u].r>>1;
	if(x<=mid)return query(u<<1,x);
	else if(x>mid)return query(u<<1|1,x);
}

int main(){
	int T;
	scanf("%d",&T);
	for(int TT=1;TT<=T;TT++){
		printf("Case #%d:\n",TT);
		
		scanf("%d",&n);
		
		memset(h,-1,sizeof h);
		memset(din,0,sizeof din);
		for(int i=1;i<n;i++){
			int a,b;
			scanf("%d%d",&a,&b);
			add(b,a);
			din[a]++;
		}
		
		timestamp=0;
		for(int i=1;i<=n;i++){
			if(!din[i])dfs(i);
		} 
		
		build(1,1,n);
		
		scanf("%d",&m);
		while(m--){
			char op[10];
			scanf("%s",op);
			if(op[0]=='C'){
				int x;
				scanf("%d",&x);
				int ans=query(1,l[x]);
				printf("%d\n",ans);
			}
			else{
				int x,k;
				scanf("%d%d",&x,&k);
				modify(1,l[x],r[x],k);
			} 
		}
	}
}

F：扫描线模板+离散化（都是基本操作）

#include <iostream>
#include <cstring>
#include <vector> 
#include <algorithm>
using namespace std;
const int N=1e5+6;
int n;
struct Segment{
	double x,y1,y2;
	int k;
	bool operator<(const Segment&t)const{
		return x<t.x;
	}
}seg[N*2];
struct node{
	int l,r;
	int cnt;
	double len; 
}tr[N*4];
vector<double>ys;

int find(double x){
	return lower_bound(ys.begin(),ys.end(),x)-ys.begin(); 
}

void push_up(int u){
	if(tr[u].cnt)tr[u].len=ys[tr[u].r+1]-ys[tr[u].l];
	else if(tr[u].l!=tr[u].r){
		tr[u].len=tr[u<<1].len+tr[u<<1|1].len;
	}
	else tr[u].len=0;
}

void build(int u,int l,int r){
	if(l==r)tr[u]={l,r,0,0};
	else{
		tr[u]={l,r};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		push_up(u);
	} 
}

void modify(int u,int l,int r,int k){
	if(tr[u].l>=l&&tr[u].r<=r){
		tr[u].cnt+=k;
		push_up(u);
	}
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid) modify(u<<1,l,r,k);
		else if(l>mid)modify(u<<1|1,l,r,k);
		else modify(u<<1,l,r,k),modify(u<<1|1,l,r,k);
		push_up(u);
	}
}

int main(){
	int T=1;
	while(~scanf("%d",&n)){
		if(n==0)break;
		for(int i=0,j=0;i<n;i++){
			double x1,y1,x2,y2;
			scanf("%lf%lf%lf%lf",&x1,&y1,&x2,&y2);
			seg[j++]={x1,y1,y2,1};
			seg[j++]={x2,y1,y2,-1};
			ys.push_back(y1),ys.push_back(y2);  
		}
		
		sort(ys.begin(),ys.end());
		ys.erase(unique(ys.begin(),ys.end()),ys.end());
		
		sort(seg,seg+n*2);
		
		build(1,0,ys.size()-2);
		
		double res=0;
		for(int i=0;i<n*2;i++){
			if(i>0)res+=tr[1].len*(seg[i].x-seg[i-1].x);
			modify(1,find(seg[i].y1),find(seg[i].y2)-1,seg[i].k); 
		}
		
		printf("Test case #%d\n",T++);
		printf("Total explored area: %.2lf\n\n",res); 
	}
}