前缀和与差分 一维 & 二维

点赞随意，欢迎点评。

前缀和与差分用于 区间修改 与 区间查询
利用递推，在O(n)时间内预计算出所有前缀和，就能利用它计算出数组中所有任意区间a[i]~a[j]的和
即：a[i] + a[i + 1] + ... + a[j - 1] + a[j] = sum[j] - sum[i - 1]
复杂度为O(n)的区间和计算 优化为O(1)的前缀和计算

一维前缀和与差分的三种操作

1. 将自身变成前缀和 b[i] = b[i] + b[i - 1]

2. 将a[i]变为b[i]的前缀和 a[i] = b[i] + a[i - 1]

3. 构造差分数组 b[i] = a[i] - a[i - 1]

前缀和与差分为便于计算求值 统一从 i = 1 或 j = 1开始计算

一维前缀和

int a[N],s[N];
for(int i = 1;i <= n;i++)
{
    cin >> a[i];
    s[i] = s[i - 1] + a[i]; // 数列递推
}
while(q--)
{
    int l,r;
    cin >> l >> r;
    cout << s[r] - s[l - 1] << endl;
}

一维差分

int a[N],b[N];
void Insert(int l,int r,int c)
{
    b[l] += c;
    b[r + 1] -= c;
}
for(int i = 1;i <= n;i++)
{
    cin >> a[i];
    Insert(i,i,a[i]); // 调用之后b[i] = a[i] - a[i - 1] 可手动模拟计算理解
}
while(q--)
{
    int l,r,c;
    scanf("%d %d %d",&l,&r,&c);
    Insert(l,r,c); // 两个Insert统一的巧妙
}
for(int i = 1;i <= n;i++)
{
    b[i] += b[i - 1];
    cout << b[i] << endl; 
}

子矩阵的和 二维前缀和

for(int i = 1;i <= n;i++)
  for(int j = 1;j <= n;j++)
  {
       cin >> a[i][j];
       s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
  }
while(q--)
{
    int x1,y1,x2,y2;
    scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
    cout << s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1] << endl;
}

差分矩阵 二维差分

void Insert(int x1,int y1,int x2,int y2)
{
    b[x1][y1] += c;
    b[x1][y2 + 1] -= c;
    b[x2 + 1][y1] -= c;
    b[x2 + 1][y2 + 1] += c;
}

for(int i = 1;i <= n;i++)
  for(int j = 1;j <= m;j++)
  {
      cin >> a[i][j];
      Insert(i,j,i,j,a[i][j]);
  }
while(q--)
{
    int x1,y1,x2,y2,c;
    scanf("%d %d %d %d %d",&x1,&y1,&x2,&y2,&c);
    Insert(x1,y1,x2,y2,c);
} 

for(int i = 1;i <= n;i++)
  for(int j = 1;j <= m;j++)
       b[i][j] += b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1];
for(int i = 1;i <= n;i++)
{
    for(int j = 1;j <= m;j++) cout << b[i][j] << endl;
    puts("");
}