蓝桥杯省赛专题训练之树状数组与线段树（相关真题和模板题）

题目：1264. 动态求连续区间和

题解：线段树模板题

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n,m;
int a[N];
int p[N<<2];
void pushup(int u){
    p[u]=p[u<<1]+p[u<<1|1];
}
void build(int u,int l,int r){
    if(l>=r){
        p[u]=a[l];
        return;
    }
    int mid=l+r>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
    pushup(u);

}
void add(int u,int l,int r,int x,int y){
    if(l>=r){
        p[u]+=y;
        return ;
    }
    int mid=l+r>>1;
    if(x<=mid) add(u<<1,l,mid,x,y);
    if(x>mid) add(u<<1|1,mid+1,r,x,y);
    pushup(u);
}
int query(int u,int l,int r,int x,int y){
    if(x<=l&&r<=y){
        return p[u];
    }
    if(x>r||y<l) return 0;
    int mid=l+r>>1;
    return query(u<<1,l,mid,x,y)+query(u<<1|1,mid+1,r,x,y);
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    build(1,1,n);
    int k,x,y;
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&k,&x,&y);
        if(k==0){
            cout<<query(1,1,n,x,y)<<endl;
        }else{
            add(1,1,n,x,y);
        }
    }
    return 0;
}

题目:1265. 数星星

题解：线段树

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n;
int p[32010<<2];
int a[15010];
void pushup(int u){
    p[u]=p[u<<1]+p[u<<1|1];
}
int query(int u,int l,int r,int x,int y){
    if(x<=l&&r<=y){
        return p[u];
    }
    if(x>r||y<l) return 0;
    int mid=l+r>>1;
    return query(u<<1,l,mid,x,y)+query(u<<1|1,mid+1,r,x,y);
}
void add(int u,int l,int r,int x){
    if(l>=r){
        p[u]++;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=x) add(u<<1,l,mid,x);
    if(mid<x) add(u<<1|1,mid+1,r,x);
    pushup(u);
}
int main(){
    cin>>n;
    int x,y;
    for(int i=0;i<n;i++){
        scanf("%d%d",&x,&y);
        a[query(1,0,32000,0,x)]++;
        add(1,0,32000,x);
    }
    for(int i=0;i<n;i++){
        printf("%d\n",a[i]);
    }
    return 0;
}

题目：1270. 数列区间最大值

题解：线段树

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n,m;
int a[N];
int p[N<<2]; 
int minn=0;
void pushup(int u){
	p[u]=max(p[u<<1],p[u<<1|1]);
}
void build(int u,int l,int r){
	if(l>=r){
		p[u]=a[l];
		return;
	}
	int mid=l+r>>1;
	build(u<<1,l,mid);
	build(u<<1|1,mid+1,r);
	pushup(u);
}
int query(int u,int l,int r,int x,int y){
	if(x<=l&&r<=y){
		return p[u];
	}
	if(x>r||y<l) return minn;
	int mid=l+r>>1;
	return max(query(u<<1,l,mid,x,y),query(u<<1|1,mid+1,r,x,y));
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
    	scanf("%d",&a[i]);
    	minn=min(minn,a[i]);
    }
    	
    build(1,1,n);
    int x,y;
    for(int i=1;i<=m;i++){
    	scanf("%d%d",&x,&y);
        printf("%d\n",query(1,1,n,x,y));
    }
    return 0;
}

题目：1215. 小朋友排队

题解：归并排序，其实每个点需要移动的最小的距离，就是归并排序中需要移动的地方。

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n;
PII a[N];
int ct[N];
void merge_sort(int l,int r){
	if(l>=r) return;
	int mid=l+r>>1;
	merge_sort(l,mid);
	merge_sort(mid+1,r);
	PII t[r-l+5];
	int k=0,i=l,j=mid+1;
	while(i<=mid&&j<=r){
		if(a[i].first<=a[j].first){
			ct[a[i].second]+=j-mid-1;
			t[k++]=a[i++];
		}else{
			ct[a[j].second]+=mid-i+1;
			t[k++]=a[j++];
		}
	}
	while(i<=mid){
		ct[a[i].second]+=r-mid;
		t[k++]=a[i++];
	}
	while(j<=r){
		t[k++]=a[j++];
	}
	for(int i=0;i<k;i++){
		a[l+i]=t[i];
	}
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
    	scanf("%d",&a[i].first);
    	a[i].second=i;
    }
    merge_sort(1,n);
    LL sum=0;
    for(int i=1;i<=n;i++){
    	//cout<<ct[i]<<" ";
    	sum+=1ll*(ct[i]+1)*ct[i]/2; 
    }
    printf("%lld",sum);
    return 0;
}

题目：1237. 螺旋折线

题解：数学找规律题。大佬思路

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int x,y;
int main(){
    cin>>x>>y;
    int k=max(abs(x),abs(y));
    if(x>=y)cout<<1ll*4*k*k+abs(x-k)+abs(y-k);
    else cout<<1ll*4*k*k-abs(x-k)-abs(y-k);
    return 0;
}

题目：797. 差分

题解：一维差分模板题

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n,m;
int a[N];
void add(int l,int r,int c){
	a[l]+=c;
	a[r+1]-=c;
}
int main(){
    cin>>n>>m;
    int x;
    for(int i=1;i<=n;i++){
    	scanf("%d",&x);
    	add(i,i,x);
    }
    int l,r,c;
    for(int i=0;i<m;i++){
    	scanf("%d%d%d",&l,&r,&c);
    	add(l,r,c);
    }
    for(int i=1;i<=n;i++){
    	a[i]+=a[i-1];
    	printf("%d ",a[i]);
    }
    return 0;
}

题目：798. 差分矩阵

题解：二维差分矩阵

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
const int N=1e5+10;
const int mod=100000007;
int n,m,q;
int a[1010][1010];
void add(int x1,int y1,int x2,int y2,int c){
	a[x1][y1]+=c;
	a[x2+1][y1]-=c;
	a[x1][y2+1]-=c;
	a[x2+1][y2+1]+=c;
}
int main(){
    cin>>n>>m>>q;
    int x;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%d",&x);
			add(i,j,i,j,x);
		}
	} 
	
	int x1,x2,y1,y2,c;
	for(int i=0;i<q;i++){
		scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&c);
		add(x1,y1,x2,y2,c);
	}
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
			printf("%d ",a[i][j]);
		}
		puts("");
	}
    return 0;
}