算法基础之前缀和

写在前面

本系列用于个人考研复试，打卡复习算法基础课知识点，以及刷一定量的题目，考研长时间没接触，代码水平下降了很多（QAQ）

前缀和

前缀和解决的问题：可以以$O(1)$的时间复杂度计算某区间，或者某区域的所有元素之和

题目一：阶段数组

题目描述

给定一个长度为 $n$的数组 $a1,a2,\dots ,an$。现在，要将该数组从中间截断，得到三个非空子数组。要求，三个子数组内各元素之和都相等。请问，共有多少种不同的截断方法？

输入格式

第一行包含整数 $n$。
第二行包含$n$个整数 $a_1,a_2,\dots ,a_n$。

输出格式

输出一个整数，表示截断方法数量。

数据范围

前六个测试点满足 $1\le n\le 10$。
所有测试点满足 $1\le n\le 10^5$，
$-10000\le a_i\le 10000$。

输入样例1：

4
1 2 3 3

输出样例1：

1

输入样例2：

5
1 2 3 4 5

输出样例2：

0

输入样例3：

2
0 0

输出样例3：

0

思路

使用前缀和数组$s$，$s[i]$表示前i个数的和，则计算$a_i$ ~ $a_j$的和为$s[j]-s[i-1]$。
枚举，考虑数据范围是$10^5$，故时间复杂度控制在$O(nlogn)$即可。
由于需要枚举两个点，而时间复杂度度此时为$O(n^2)$，不可行，所以考虑枚举一个点。

代码如下

#include <iostream>
#include <cstring>
#include <algorithm>

typedef long long LL;

using namespace std;

const int N = 100010;

int q[N], s[N];
int n;

int main()
{
    cin >> n;
    int x;
    for (int i = 1; i <= n; i ++)
    {
        cin >> x;
        s[i] = s[i - 1] + x;

    }
    LL res = 0;

    if(s[n] % 3 != 0) res = 0;
    else {
        LL cnt = 0;
        for (int j = 2; j < n; j ++ )
        {
            if (s[j - 1] == s[n] / 3) cnt ++ ;
            if (s[j] == s[n] / 3 * 2) res += cnt;
        }
    }
    cout << res << endl;

    return 0;
}

题目二：前缀和

题目描述

输入一个长度为 $n$ 的整数序列。
接下来再输入 $m$ 个询问，每个询问输入一对 $l,r$。
对于每个询问，输出原序列中从第 $l$ 个数到第 $r$ 个数的和。

输入格式

第一行包含两个整数 $n$ 和 $m$。

第二行包含 $n$ 个整数，表示整数数列。

接下来 $m$ 行，每行包含两个整数 $l$ 和 $r$，表示一个询问的区间范围。

输出格式

共 $m$ 行，每行输出一个询问的结果。

数据范围

$1\le l\le r\le n$,
$1\le n,m\le 100000$,
$-1000\le 数列中元素的值\le 1000$

输入样例：

5 3
2 1 3 6 4
1 2
1 3
2 4

输出样例：

3
6
10

思路

这题模板题，使用前缀和数组$s$，$s[i]$表示前i个数的和，则计算$a_i$ ~ $a_j$的和为$s[j]-s[i-1]$。

代码

#include <iostream>

using namespace std;

const int N = 100010;

int s[N];
int n, m;

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++)
    {
        int x;
        cin >> x;
        s[i] = s[i - 1] + x;
    }

    while (m -- ) 
    {
        int l, r;
        cin >> l >> r;
        cout << s[r] - s[l - 1] << endl;

    }
    return 0;
}

题目三：子矩阵的和

题目描述

输入一个 $n$ 行 $m$ 列的整数矩阵，再输入 $q$ 个询问，每个询问包含四个整数 $x1,y1,x2,y2$，表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。

输入格式

第一行包含三个整数 $n，m，q$。

接下来 $n$ 行，每行包含 $m$ 个整数，表示整数矩阵。

接下来 $q$ 行，每行包含四个整数 $x1,y1,x2,y2$，表示一组询问。

输出格式

共 $q$ 行，每行输出一个询问的结果。

数据范围

$1\le n,m\le 1000$,
$1\le q\le 200000$,
$1\le x1\le x2\le n$,
$1\le y1\le y2\le m$,
$-1000\le 矩阵内元素的值\le 1000$

输入样例：

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

输出样例：

17
27
21

思路

本题二维前缀和模板题

代码

#include <iostream>

using namespace std;

const int N = 1010;

int s[N][N];
int n, m, q;

int main()
{
    cin >> n >> m >> q;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++)
        {
            int x;
            cin >> x;
            s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + x;
        }
    
    while (q -- )
    {
        int x1, y1, x2, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        cout << s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1] << endl;
    }
    
    return 0;
    
}

题目四：K倍区间

题目描述

给定一个长度为 $N$ 的数列，$A_1,A_2,\dots A_N$，如果其中一段连续的子序列 $A_i,A_{i+1},\dots A_j$ 之和是 $K$ 的倍数，我们就称这个区间 $[i,j]$ 是 $K$ 倍区间。

你能求出数列中总共有多少个 $K$ 倍区间吗？

输入格式

第一行包含两个整数 $N$ 和 $K$。

以下 $N$ 行每行包含一个整数 $A_i$。

输出格式

输出一个整数，代表 $K$ 倍区间的数目。

数据范围

$1\le N,K\le 100000$,
$1\le A_i\le 100000$

输入样例：

5 2
1
2
3
4
5

输出样例：

6

思路

记$S_i=A_1+A_2+...+A_i$
则任意区间 $[i,j]$ 子序列之和可以表示为$S_j-S_{i-1}$
要求$(S_j-S_{i-1})$% $k==0$ 有多少个
则等价于 $S_j$ % $k==$ $S_{i-1}$ % $k$ 有多少个
当我们算第$i$个数时，计算$S_0$ ~ $S_{i-1}$ 中有多少个 % $k==S_i$ % $k$
于是用数组开哈希表$cnt[x]$ 表示$S_0$ ~ $S_N$ 中有多少个余数为$x$的数
使用哈希表将二维优化成一维

本题思维难度较大，很经典的蓝桥杯题目

代码

#include <iostream>

using namespace std;

typedef long long LL;

const int N = 100010;

int n, k;
LL s[N];
int cnt[N];

int main()
{
    cin >> n >> k;
    
    for (int i = 1; i <= n; i ++)
    {
        int x;
        cin >> x;
        s[i] = s[i - 1] + x;
    }
    cnt[0] ++;
    LL res = 0;
    for (int i = 1; i <= n; i ++)
    {
        res += cnt[s[i] % k];
        cnt[s[i] % k] ++;
    }

    cout << res << endl;
    return 0;
}

题目五：激光炸弹

题目描述

地图上有 $N$ 个目标，用整数 $X_i,Y_i$ 表示目标在地图上的位置，每个目标都有一个价值 $W_i$。

注意：不同目标可能在同一位置。

现在有一种新型的激光炸弹，可以摧毁一个包含 $R\times R$ 个位置的正方形内的所有目标。

激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆炸范围，即那个正方形的边必须和 $x，y$ 轴平行。

求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式

第一行输入正整数 $N$ 和 $R$，分别代表地图上的目标数目和正方形包含的横纵位置数量，数据用空格隔开。

接下来 $N$ 行，每行输入一组数据，每组数据包括三个整数 $Xi,Yi,Wi$，分别代表目标的 $x$ 坐标，$y$ 坐标和价值，数据用空格隔开。

输出格式

输出一个正整数，代表一颗炸弹最多能炸掉地图上目标的总价值数目。

数据范围

$0\le R\le 109$,
$0<N\le 10000$,
$0\le X_i,Y_i\le 5000$,
$0\le W_i\le 1000$

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

思路

使用二维前缀和
范围R只能炸毁[R - 1] * [R - 1]范围的价值
矩阵枚举范围为R - 1

代码

#include <iostream>

using namespace std;

const int N = 5010;

int g[N][N];

int n, r;

int main()
{
    cin >> n >> r;
    r = min(r, 5001);
    for (int i = 0; i < n; i ++)
    {
        int x, y, w;
        cin >> x >> y >> w;
        x ++, y ++;
        g[x][y] += w;
    }
    
    for (int i = 1; i <= 5001; i ++) 
        for (int j = 1; j <= 5001; j ++)
            g[i][j] += g[i - 1][j] + g[i][j - 1] - g[i - 1][j - 1]; 
    
    int res = 0;
    for (int i = r; i <= 5001; i ++)
        for (int j = r; j <= 5001; j ++)
            res = max(res, g[i][j] - g[i - r][j] - g[i][j - r] + g[i - r][j - r]);
            
    cout << res << endl;
    
    return 0;
}

完