前缀和与差分 一维 & 二维

点赞随意,欢迎点评

前缀和与差分用于 区间修改 与 区间查询
利用递推,在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]

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

  1. 构造差分数组 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("");
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值