前缀和（简讲）

一·定义

前缀和可以简单理解为「数列的前  项的和」，是一种重要的预处理方式，能大大降低查询的时间复杂度。C++ 标准库中实现了前缀和函数 std::partial_sum，定义于头文件 <numeric> 中。

例题

有 n个的正整数放到数组a 里，现在要求一个新的数组b，新数组的第i个数 b[i] 是原数组 a 第 0到第i个数的和

输入：

5

1 2 3 4 5

输出：

1 3 6 10 15

解题思路

递推：B[0] = A[0]，对于i>=1 则 B[i] = B[i-1] + A[i]。

代码实现

#include <iostream>
using namespace std;
int N, A[10000], B[10000];
int main() {
  cin >> N;
  for (int i = 0; i < N; i++) {
    cin >> A[i];
  }
  // 前缀和数组的第一项和原数组的第一项是相等的。
  B[0] = A[0];
  for (int i = 1; i < N; i++) {
    // 前缀和数组的第 i 项 = 原数组的 0 到 i-1 项的和 + 原数组的第 i 项。
    B[i] = B[i - 1] + A[i];
  }
  for (int i = 0; i < N; i++) {
    cout << B[i] << " ";
  }
  return 0;
}

二·二维、多维前缀和

多维前缀和的普通求解方法几乎都是基于容斥原理。

例题

见洛谷 P1387 最大正方形

实现代码

#include <algorithm>
#include <iostream>
using namespace std;
int a[103][103];
int b[103][103];  // 前缀和数组，相当于上文的 sum[]
int main() {
  int n, m;
  cin >> n >> m;
  for (int i = 1; i <= n; i++) {
    for (int j = 1; j <= m; j++) {
      cin >> a[i][j];
      b[i][j] =
          b[i][j - 1] + b[i - 1][j] - b[i - 1][j - 1] + a[i][j];  // 求前缀和
    }
  }
  int ans = 1;
  int l = 2;
  while (l <= min(n, m)) {  // 判断条件
    for (int i = l; i <= n; i++) {
      for (int j = l; j <= m; j++) {
        if (b[i][j] - b[i - l][j] - b[i][j - l] + b[i - l][j - l] == l * l) {
          ans = max(ans, l);  // 在这里统计答案
        }
      }
    }
    l++;
  }
  cout << ans << endl;
  return 0;
}