二维差分数组: 每次修改区域为 O ( 1 ) O(1) O(1),每次查询为 O ( n m ) O(nm) O(nm)。
例题:P3397 地毯
#include<bits/stdc++.h>
#define ll long long
#define lf double
#define IOS std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define DEBUG cout<<",here\n";
#define Rep(i,l,r) for(int i=(l);i<=(r);i++)
#define rep(i,l,r) for(int i=(l);i< (r);i++)
#define Rev(i,r,l) for(int i=(r);i>=(l);i--)
#define rev(i,r,l) for(int i=(r);i> (l);i--)
#define MAXN 1005
using namespace std;
struct Dif_2D
{
const static int maxn=MAXN;
int sum[maxn][maxn];
int a[maxn][maxn];
void add(int x1,int y1,int x2,int y2,int num){//以(x1,y1)为左上角,(x2,y2)为右下角的区域加上sum
a[x1][y1]+=num;a[x1][y2+1]-=num;
a[x2+1][y1]-=num;a[x2+1][y2+1]+=num;
}
void solve(int n,int m){//每次查询前要执行一次
Rep(i,1,n){
Rep(j,1,m){
sum[i][j]=a[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
}
}
}
int get_point(int x,int y){return sum[x][y];}//获取一个点的值
int get_sum(int x1,int y1,int x2,int y2){//获取一个区域的和
int ans=sum[x2][y2]-sum[x2][y1-1]-sum[x1-1][y2]+sum[x1-1][y1-1];
return ans;
}
};Dif_2D a;
int n,m;
int main()
{
cin>>n>>m;
Rep(i,1,m){
int x1,y1,x2,y2;
cin>>x1>>y1>>x2>>y2;
a.add(x1,y1,x2,y2,1);
}
a.solve(n,n);
Rep(i,1,n){
Rep(j,1,n){
cout<<a.get_point(i,j)<<" ";
}cout<<"\n";
}
return 0;
}
参考:
一、二维差分与前缀和与例题汇总