二维前缀和
功能:
计算一个 a × b 的矩阵里各个元素的总和
计算方法:
若A、B、C 均为(1,1)到其位置的矩阵的元素总和,
(d 为 D 位置上的元素)
则
D = B + C - A + d ;
代码实现
for (int i = 1; i <=a; ++i) {
for (int j = 1; j <= b; ++j) {
m[i][j]+=m[i-1][j]+m[i][j-1]-m[i-1][j-1];
}
}
二维差分
功能
计算矩阵(a,b) - (a1,b1) 里各个元素的和
计算方法:
若 A、C、B、D 均为(1,1)到其位置的矩阵的所有元素总和
要计算 A - D 矩阵的元素总和
则
ans = D - B - C + E
代码实现
for (int i = a; i <=a1; ++i) {
for (int j = b; j <=b1; ++j) {
ans = max(ans,m[i][j]-m[i-a][j]-m[i][j-b]+m[i-a][j-b]);
}
}
练习题
二哥种花生
Description
二哥在自己的后花园里种了一些花生,也快到了收获的时候了。这片花生地是一个长度为L、宽度为W的矩形,每个单位面积上花生产量都是独立的。他想知道,对于某个指定的区域大小,在这么大的矩形区域内,花生的产量最大会是多少。
Input Format
第1行有2个整数,长度L和宽度W。
第2行至第L+1行,每行有W个整数,分别表示对应的单位面积上的花生产量A( 0 ≤ A < 10)。
第L+2行有2个整数,分别是指定的区域大小的长度a和宽度b。
Output Format
输出一个整数m,表示在指定大小的区域内,花生最大产量为m。
Sample Input
4 5
1 2 3 4 5
6 7 8 0 0
0 9 2 2 3
3 0 0 0 1
3 3
Sample Output
38
样例解释
左上角:38 = (1+2+3) + (6+7+8) + (0+9+2)
数据范围
对于30%的数据: 1 ≤ L , W ≤ 100 ;
对于100%的数据: 1 ≤ L , W ≤ 1000。
全部区域大小满足:1 ≤ a ≤ L,1 ≤ b ≤ W 。
code
#include <iostream>
using namespace std;
int main(){
int l,w;
cin >> l >> w;
int m[l+3][w+3];
for (int i = 1; i <= l; ++i) {
for (int j = 1; j <= w; ++j) {
cin >> m[i][j];
}
}
for (int i = 1; i <= l; ++i) {
for (int j = 1; j <= w; ++j) {
m[i][j]+=m[i-1][j]+m[i][j-1]-m[i-1][j-1];
}
}
int a,b;
int ans=0;
cin >> a >> b;
for (int i = a; i <=l; ++i) {
for (int j = b; j <=w; ++j) {
ans = max(ans,m[i][j]-m[i-a][j]-m[i][j-b]+m[i-a][j-b]);
}
}
cout << ans;
return 0;
}