2020 8.7 暑期线段树专题

线段树可以在 n*logn 的时间复杂度内实现单点修改、区间 修改、区间查询（区间求和，求区间最大值，求区间最小值） 等操作。

 

 

 

 

 

 

在这里顺便把树状数组附上，树状数组只能区间求和时间复杂度也是n*log n

 

 

code:

 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, m;
int a[N], tr[N];

int lowbit(int x)
{
    return x & -x;
}

void add(int x, int v)//这个原理很烦，但是你休息只有后面的树状数组才会包含x 
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += v;
}

int query(int x)//其实lowbit(i)就表示tr[i]所包含的元素个数 
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i ++ ) add(i, a[i]);

    while (m -- )
    {
        int k, x, y;
        scanf("%d%d%d", &k, &x, &y);
        if (k == 0) printf("%d\n", query(y) - query(x - 1));
        else add(x, y);
    }

    return 0;
}

 

 

 

B - 敌兵布阵

 

模板维护区间和

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0x3f3f3f3f

using namespace std;

const int N=2e5+10;
int w[N];
struct node{
	int l,r;
	int maxv;
}tr[4*N]; 
int n,m;

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

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

int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].maxv;
	else{
		int mid=tr[u].l+tr[u].r>>1;
		int maxx=-INF;
		if(l<=mid) maxx =max(maxx, query(u<<1,l,r));
		if(r>mid) maxx =max(maxx,query(u<<1|1,l,r));
		return maxx;
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m)){
		for(int i=1;i<=n;i++) scanf("%d",&w[i]);
		int a,b;
		char ch;
		build(1,1,n);
		while(m--){
			scanf(" %c%d%d",&ch,&a,&b);//%c和%d同时输入时%c前面加个空格 
			if(ch=='U') modify(1,a,b);
		 	else printf("%d\n",query(1,a,b));
		}
	}
	return 0;
 } 

 

D - I Hate It

模板维护区间最大值

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0x3f3f3f3f

using namespace std;

const int N=2e5+10;
int w[N];
struct node{
	int l,r;
	int maxv;
}tr[4*N]; 
int n,m;

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

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

int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].maxv;
	else{
		int mid=tr[u].l+tr[u].r>>1;
		int maxx=-INF;
		if(l<=mid) maxx =max(maxx, query(u<<1,l,r));
		if(r>mid) maxx =max(maxx,query(u<<1|1,l,r));
		return maxx;
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m)){
		for(int i=1;i<=n;i++) scanf("%d",&w[i]);
		int a,b;
		char ch;
		build(1,1,n);
		while(m--){
			scanf(" %c%d%d",&ch,&a,&b);//%c和%d同时输入时%c前面加个空格 
			if(ch=='U') modify(1,a,b);
		 	else printf("%d\n",query(1,a,b));
		}
	}
	return 0;
 } 

C - Billboard

 

1到h的叶子结点就相当于从高度1到高度h

 

 

code：

#include<iostream>
#include<cstdio> 

using namespace std;

const int N =2e5+10;

struct node{
	int l,r;
	int maxv;
}tr[4*N];//tr[1].maxv表示1到h的叶子节点的最大值 
int h,w,n; 

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

int query(int u,int x){
	int res=-1;
	if(tr[u].l==tr[u].r&&tr[u].maxv>=x){//注意这个条件，h有可能是1 
		tr[u].maxv-=x;
		res=tr[u].l;
	}
	else{
		if(tr[u<<1].maxv>=x) res=query(u<<1,x);
		else if(tr[u<<1|1].maxv>=x) res=query(u<<1|1,x);
		tr[u].maxv=max(tr[u<<1].maxv,tr[u<<1|1].maxv);
	}
	return res;
} 
int main()
{
	while(~scanf("%d%d%d",&h,&w,&n)){
		if(h>=n){
			h=n;//h个叶子节点最多只会用到n个 
		}
		build(1,1,h);
		int x;
		for(int i=1;i<=n;i++){
			scanf("%d",&x);
			printf("%d\n",query(1,x));
		} 
	}
	return 0;
}

 

E - Balanced Lineup

 

 思路：求区间最小值加上最大值

code：

 

#include<iostream>
#include<cstdio> 

using namespace std;

const int N =5e4+10;

struct node{
	int l,r;
	int maxv,minv;
}tr[4*N];//tr[1].maxv表示1到h的叶子节点的最大值 
int n,q;
int w[N];

void build(int u,int l,int r){
	if(l==r){
		tr[u]={l,r,w[l],w[l]};
	}
	else{
		tr[u]={l,r};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		tr[u].maxv=max(tr[u<<1].maxv,tr[u<<1|1].maxv);
		tr[u].minv=min(tr[u<<1].minv,tr[u<<1|1].minv);
	}
}

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

int query2(int u,int l,int r){
	if(l<=tr[u].l&&r>=tr[u].r){
		return tr[u].minv;		
	}
	else{
		int minx=2e6+10;
		int mid= tr[u].l+tr[u].r>>1;
		if(l<=mid) minx=min(minx,query2(u<<1,l,r));
		if(r>mid) minx=min(minx,query2(u<<1|1,l,r)); 
		return minx;
	}
} 
int main()
{
	scanf("%d%d",&n,&q);
	for(int i =1;i<=n;i++){
		scanf("%d",&w[i]);
	}
	build(1,1,n);
	int a,b;
	while(q--){
		scanf("%d%d",&a,&b);
		printf("%d\n",query1(1,a,b)-query2(1,a,b));
	}	
	return 0;
}