二维前缀和主要用来求一个矩形区域内的数量和。
如下图所示,我们假设dp[i][j]为矩形 (1,1)到(i, j)之间区域的面积,则:
dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1]+mp[i][j]
其中mp[i][j]为(i, j)点的大小,少掉了一个蓝色部分的区域,这里就需要加上mp[i][j]。
以上步骤用来构建二维前缀和。然后根据实际情况来进行运用,以下以计蒜客中的天上的星星这个题为例:
题目中给出的区间是一个矩形区域,让我们求这个矩形区域中星星的亮度总和,根据以上步骤首先构造出二维前缀和,之后题目中输入的x1, y1, x2, y2为矩形的对角线,对应到数组中为矩形的左上角和右下角。因为题目中说边上的星星也要计算。所以我们的状态方程为:
res = dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1]
AC代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 2005
int star[maxn][maxn], dp[maxn][maxn];
int main(){
int n;
cin>>n;
int x, y, w;
memset(star, 0, sizeof(star));
memset(dp, 0, sizeof(dp));
while(n--){
cin>>x>>y>>w;
dp[x+1][y+1] += w; //下标从1开始,避免下边减一之后为负数
}
for(int i = 1; i < maxn; i++){
for(int j = 1; j < maxn; j++){
dp[i][j] = dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+dp[i][j];
}
}
int q; cin>>q;
while(q--){
int x0, y0, x1, y1;
cin>>x0>>y0>>x1>>y1;
cout<<dp[x1+1][y1+1]-dp[x0][y1+1]-dp[x1+1][y0]+dp[x0][y0]<<endl;
}
return 0;
}