三种区间操作：差分前缀和、线段树和树状数组

区间操作

区间操作的思路有：差分、前缀和、线段树和树状数组

在更低的时间复杂度下完成一个连续区域的加减问题

差分和前缀和

洛谷 P3397 地毯
#include<bits/stdc++.h>//二维差分和前缀和
using namespace std;
const int N=2000;
int n,t,a[N][N];

void print()
{
	for(int i=1;i<n+1;i++)
	{
		for(int j=1;j<n+1;j++)
		{
			cout<<a[i][j]<<" ";
		}
		cout<<endl;
	}
}

int main()
{
	cin>>n>>t;
	int ax,ay,bx,by;
	for(int i=0;i<t;i++)
	{
		cin>>ax>>ay>>bx>>by;//构建差分数组
		a[ax][ay]++;
		a[ax][by+1]--;
		a[bx+1][by+1]++;
		a[bx+1][ay]--; 
	}
	for(int i=1;i<n+1;i++)
	{
		for(int j=0;j<n+1;j++)
		{
			a[i][j]+=a[i-1][j];
		}
	}
	for(int i=0;i<n+1;i++)
	{
		for(int j=1;j<n+1;j++)
		{
			a[i][j]+=a[i][j-1];//做前缀和
		}
	}
	print();
	return 0;
}

洛谷 P1719 最大加权矩形
#include<iostream>//因为题目n<120，O(n^4)可以做
using namespace std;
int n;
int a[130][130];//存储题目中的矩阵
int sum[130][130];
int qz[130][130];//qz[i][j]指的是第i行到j的前缀和
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>a[i][j];
            qz[i][j]=qz[i][j-1]+a[i][j];//求前缀和
            sum[i][j]=qz[i][j]+sum[i-1][j];//计算sum
        }
    }
    int mx=-1e9;//存储答案
    for(int x1=1;x1<=n;x1++){
    	for(int y1=1;y1<=n;y1++){
    		for(int x2=1;x2<=n;x2++){
    			for(int y2=1;y2<=n;y2++){
    				if(x2<x1 || y2<y1) continue;//如果左上角比右下角还要大，就不用求了，下一个
    				mx=max(mx,sum[x2][y2]+sum[x1-1][y1-1]-sum[x2][y1-1]-sum[x1-1][y2]);//求最大值
    			}
    		}
    	}
    }
    cout<<mx;//输出
	return 0;
}

线段树

#include<bits/stdc++.h>
using namespace std;
#define int unsigned long long

const int N=2e5+10;
int tree[N*2],a[N],n,process,mark[N*2];//mark是懒标记 

void build(int f,int e,int p);
void strength_print_tree();
void add(int x,int y,int k,int f,int e,int p);
int search(int x,int y,int f,int e,int p);

signed main()
{
	cin>>n>>process;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	build(1,n,1);
	while(process--)
	{
		int num;
		cin>>num;
		if(num==1)
		{
			int x,y,k;
			cin>>x>>y>>k;
			add(x,y,k,1,n,1);
		}
		else
		{
			int x,y;
			cin>>x>>y;
			cout<<search(x,y,1,n,1)<<endl;
		}
	}
	//strength_print_tree();
	return 0;
}

void build(int f,int e,int p)
{
	if(f==e)
	{
		tree[p]=a[f];
	}
	else
	{
		int mid=(f+e)/2;
		build(f,mid,p*2);
		build(mid+1,e,p*2+1);
		tree[p]=tree[p*2]+tree[p*2+1];
	}
}
void strength_print_tree()
{
	int f=1,e=n*2;
	for(int i=f;i<=e;i++)
	{
		cout<<"tree中第"<<i<<"为"<<tree[i]<<endl;
	}
}
void add(int x,int y,int k,int f,int e,int p)//f,e是树的，x,y为定区间
{
	if(y<f||x>e)
	{
		return;
	}
	else if(x<=f&&y>=e)
	{
		tree[p]+=(e-f+1)*k;
		if(f<e) mark[p]+=k;
	}
	else
	{
		int mid=(f+e)/2;
		mark[p*2]+=mark[p];//懒标记 
		mark[p*2+1]+=mark[p];
		tree[p*2]+=mark[p]*(mid-f+1);
		tree[p*2+1]+=mark[p]*(e-mid);
		mark[p]=0;
		add(x,y,k,f,mid,p*2);
		add(x,y,k,mid+1,e,p*2+1);
		tree[p]=tree[p*2]+tree[p*2+1];
	}
}
int search(int x,int y,int f,int e,int p)
{
	if(x>e||y<f)
	{
		return 0;
	}
	else if(x<=f&&y>=e)
	{
		return tree[p];
	}
	else
	{
		int mid=(f+e)/2;
		mark[p*2]+=mark[p];
		mark[p*2+1]+=mark[p];
		tree[p*2]+=mark[p]*(mid-f+1);
		tree[p*2+1]+=mark[p]*(e-mid);
		mark[p]=0;
		return search(x,y,f,mid,p*2)+search(x,y,mid+1,e,p*2+1);
	}
}

树状数组

支持操作：单点改值，区间查询；区间改值，单点查值（就是前缀和）；区间改值，区间查值

#include<bits/stdc++.h>
using namespace std;

const int N=5e5+10;
int n,m,c[N],p,x,y,a;

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

void add(int t,int a)
{
	for(int j=t;j<=n;j+=lowbit(j))
	{
		c[j]+=a;
	}
}

int sumt(int u)
{
	int ans=0;
	for(int i=u;i>0;i-=lowbit(i))
	{
		ans+=c[i];
	}
	return ans;
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>a;
		add(i,a);//单点改值
	}
	for(int i=1;i<=m;i++)
	{
		cin>>p>>x>>y;
		if(p==1)
		{
			add(x,y);
		}
		else
		{
			cout<<sumt(y)-sumt(x-1)<<endl;
		}
//		cout<<"test"<<endl;
//		for(int i=1;i<=n;i++)
//		{
//			cout<<c[i]<<" "; 
//		}
//		cout<<"\n";
			}
	return 0;
}

#include<bits/stdc++.h>
using namespace std;
//把前i项的和作为第i项的数字 
const int N=5e5+10;
int n,m,c[N],p,x,y,a;

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

void add(int t,int a)
{
	for(int j=t;j<=n;j+=lowbit(j))
	{
		c[j]+=a;
	}
}

int sumt(int u)
{
	int ans=0;
	for(int i=u;i>0;i-=lowbit(i))
	{
		ans+=c[i];
	}
	return ans;
}

int main()
{
	cin>>n>>m;
	int last=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a;
		add(i,a-last);
		last=a;
	}
	for(int i=1;i<=m;i++)
	{
		cin>>p;
		if(p==1)
		{
			cin>>x>>y>>a;
			add(x,a);
			if(y!=n) add(y+1,-a);
		}
		else
		{
			cin>>x;
			cout<<sumt(x)<<endl;
		}
	}
	return 0;
}