【基础算法】区间处理技巧（前缀和及差分）

石韫玉而山辉

于 2024-01-28 22:23:15 发布

阅读量626

点赞数 12

分类专栏：数据结构与算法文章标签：算法数据结构

本文链接：https://blog.csdn.net/m0_71146126/article/details/135901703

版权

数据结构与算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

1. 前缀和

1.1. 一维前缀和

前缀和 $S_{i} = a_1+a_2+...+a_i$

计算公式：

S[i] = a[1] + a[2] + ... a[i]

a[l] + ... + a[r] = S[r] - S[l - 1]

sum[0]=0;
for(int i=1;i<=n;i++)
    sum[i]=sum[i-1]+a[i];

相关题目

洛谷：【深进1.例1】求区间和 - 洛谷

AcWing：795. 前缀和 - AcWing题库

1.2. 二维前缀和

S[i, j] = 第i行j列格子左上部分所有元素的和

以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：

S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

for(int i=1; i<=n; i++)
    for(int j=1; j<=m; j++)
        sum[i][j] = sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1] + a[i][j];

相关题目

AcWing：796. 子矩阵的和 - AcWing题库

2. 差分

2.1. 一维差分

差分数组 diff[i]=a[i]-a[i-1]

适用场合：频繁对数组的某个区间的元素进行删减

例如：给区间[l, r]中的每个数加上d【使用差分可以实现O(1)】

diff[l]+=d;

if(j<n) diff[r+1]-=d;

diff[1]=a[1]; //计算差分数组
for(int i=2;i<=n;i++)
diff[i]=a[i]-a[i-1];
/*一系列操作*/
a[1]=diff[1]; //利用diff反推/恢复数组a
for(int i=2; i<=n; i++)
a[i]=a[i-1]+diff[i];

/* 给定一个长度为n的数组a（下标从1开始），做m次操作，每次操作
是对区间i至j的所有元素加上d，返回最后的数组。
预处理O(n) 每次区间操作O(1) 时间复杂度O(m+n)*/
const int N=5e6+10;
void RangeIncrement(int a[],int n,int m){
    int diff[N],i,j,d;
    diff[1]=a[1]; //计算差分数组
    for(int i=2;i<=n;i++)
        diff[i]=a[i]-a[i-1];
    while(m--){ //a[i]…a[j]加d
        scanf(“%d %d %d”,&i,&j,&d);
        diff[i]+=d;
        if(j<n) diff[j+1]-=d;
    }
    a[1]=diff[1]; //利用diff反推/恢复数组a
    for(int i=2; i<=n; i++)
        a[i]=a[i-1]+diff[i];
}