二维数组前缀和对于求二维数组求子区间的和能极大的缩短时间复杂度
主要原理是:
S[i, j] = 第i行j列格子左上部分所有元素的和(s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1])
以(x1, y1)为左上角,(x2, y2)为右下角的子区间的和为:
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]
例如:求二维数组
1 2 4 3
5 1 2 4
6 3 5 9
的子区间的和
#include<iostream>
using namespace std;
int s[110][110];
int main()
{
for (int i = 1; i <= 3; i++)
{
for (int j = 1; j <= 4; j++)
{
cin >> s[i][j];//先保存二维数组
}
}
for (int i = 1; i <= 3; i++)
{
for (int j = 1; j <= 4; j++)
{
s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];//求前缀和公式
}
}
int x1, y1, x2, y2;
cin >> x1 >> y1 >> x2 >> y2;
cout << s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1];//根据公式求子区间的和
}
输入:
1 2 4 3
5 1 2 4
6 3 5 9
1 2 2 3
输出结果: