前缀和与差分

前缀和

一维前缀和的维护公式为：s[i]=s[i-1]+a[i]。
二维前缀和的维护公式为：s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j]

#include<iostream>
using namespace std;
int get_sum(int* sum,int L,int R){
	if(L<=0) return sum[R];
	return sum[R]-sum[L-1]; 
}
int main(){
	int n,q;
	cin>>n>>q;
	int* a=new int[n];
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	int* sum=new int [n];
	sum[0]=a[0];
	for(int i=1;i<n;i++){
		sum[i]=sum[i-1]+a[i];
		
	}
	for(int i=0;i<n;i++){
		cout<<sum[i]<<" ";
	}
	cout<<endl;
	while(q--){
		cout<<"请输入查询的L和R"<<endl;
		int L,R;
		cin>>L>>R;
		cout<<get_sum(sum,L,R);
	}
	return 0;
}

差分

一维差分数组：b[i]=a[i]-a[i-1]。
二维差分数组：b[i][j]=a[i][j]-a[i][j-1]-(a[i-1][j]-a[i-1][j-1])=a[i][j]-a[i][j-1]-a[i-1][j]+a[i-1][j-1]。

#include<iostream>
using namespace std;
void mode(int* d,int L,int R,int e){
	d[L]+=e;
	d[R+1]-=e;
}
int main(){
	int n,m;
	cin>>n>>m;
	int* a=new int[n+1];
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	int* d=new int [n+1];

	for(int i=0;i<n;i++){
		d[i]=0;
		
	}
	
	
	while(m--){
		cout<<"操作的L和R以及数"<<endl;
		int L,R,e;
		cin>>L>>R>>e;
		mode(d,L,R,e);
	}
	
	int* sum=new int [n];
		sum[0]=d[0];
		for(int i=1;i<n;i++){
			sum[i]=sum[i-1]+d[i];
			
		}
	for(int i=0;i<n;i++){
		cout<<a[i]<<" "; 
		a[i]+=sum[i];
		}
		
	return 0;
}