算法专题篇四:前缀和

算法专题篇四:前缀和_第1张图片

"回忆里的我,比国王富有。奢侈的快乐~" 


 1、前缀和【模板】

(1) 题目解析

算法专题篇四:前缀和_第2张图片

(2) 算法原理         算法专题篇四:前缀和_第3张图片

#include 
using namespace std;

const int N = 100010;
// 可能出现溢出
long long arr[N],dp[N];
int n,q;

int main() {
    cin >> n >> q;
    // 初始化数组
    for(int i=1;i<=n;++i) cin >> arr[i];

    // 创建dp
    for(int i = 1; i <= n; i++) dp[i] = dp[i - 1] + arr[i];

    // 查询
    while(q--){
        int l,r;
        cin >> l >> r;
        // 计算区间和
        cout << dp[r] - dp[l - 1] << endl;
    }

    return 0;
}

 2、DP35 【模板】二维前缀和

(1) 题目解析

算法专题篇四:前缀和_第4张图片

        同一维前缀和类似,这类题型都需要预处理出前缀和数组,再求和。

(2) 算法原理         

算法专题篇四:前缀和_第5张图片

#include 
using namespace std;
const int N = 1010;
int main() {
    int n, m, q;
    long long arr[N][N], dp[N][N];
    cin >> n >> m >> q;
    // 输入数组
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
            cin >> arr[i][j];

    // 预处理前缀和
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
            dp[i][j] = dp[i - 1][j] + dp[i][j - 1] + arr[i][j] - dp[i - 1][j - 1];

    // 查询
    int x1,y1,x2,y2;
    while(q--)
    {
        cin >> x1 >> y1 >> x2 >> y2;
        cout << dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1] << endl;
    }

    return 0;
}

3、寻找数组的中心下标

(1) 题目解析

算法专题篇四:前缀和_第6张图片

        本题的核心在于,存在一个mid,使得sum(0,mid-1) == sum(mid+1,end)。所以,我们只需要求得 (0,mid-1) 区间的值 是否等于 (mid+1,end)的值,就可以判断该mid 是否是中心下标。

(2) 算法原理         算法专题篇四:前缀和_第7张图片

class Solution {
public:
    int pivotIndex(vector& nums) {
        int n = nums.size();
        vector f(n+1);
        vector g(n+1);

        // 初始化前缀和
        f[0] = g[n-1] = 0;
        for(int i =1;i<=n;++i) f[i] = f[i-1] + nums[i-1];
        for(int i =n-2;i>=0;--i) g[i] = g[i+1] + nums[i+1];

        // 判断 0~n-1 迭代下标
        for(int i=0;i

4、除自身以外数组的乘积

(1) 题目解析        算法专题篇四:前缀和_第8张图片

        

(2) 算法原理        算法专题篇四:前缀和_第9张图片

class Solution {
public:
    vector productExceptSelf(vector& nums) {
        int n = nums.size();
        vector f(n+1);
        vector g(n+1);

        // 初始化前缀和
        f[0] = g[n-1] = 1;
        for(int i=1;i<=n;++i) f[i] = f[i-1]*nums[i-1];
        for(int i=n-2;i>=0;--i) g[i] = g[i+1]*nums[i+1];

        // 判断
        vector ret;
        for(int i=0;i

5、和为 K 的子数组

(1) 题目解析

算法专题篇四:前缀和_第10张图片

        什么是子数组? 子数组一定是连续的 --> 这本质对应的是前缀和因为前缀和就是连续区间。

(2) 算法原理         算法专题篇四:前缀和_第11张图片

class Solution {
public:
    int subarraySum(vector& nums, int k) { 
        unordered_map hash; 
        hash[0] = 1;  // 细节处理
        int sum = 0; // 模拟前缀和
        int count = 0;
        for(auto& num:nums)
        {
            sum += num;
            if(hash.count(sum-k)){
                // 该元素存在
                // 有多少个这个元素 就能找出多少个k
                count += hash[sum-k];
            }  
            // 该元素值个数++
            hash[sum]++;
        }
        return count;
    }
};

6、和可被 K 整除的子数组

(1) 题目解析        算法专题篇四:前缀和_第12张图片

         相比于上题,这道题无非条件做了些改变,在数组里找到能被k整除的前缀和。不过在开始这道题之前,我们得补充另外一个知识。

同余定理:

        给定一个正整数m,如果两个整数a和b满足a-b能够被m整除,即(a-b)/m得到一个整数,那么就称整数a与b对模m同余,记作a≡b(mod m)。对模m同余是整数的一个等价关系。

        简单来说,就是这样:

算法专题篇四:前缀和_第13张图片

        有了上述的基础,我们再来认识认识这道题。

(2) 算法原理        

算法专题篇四:前缀和_第14张图片

class Solution {
public:
    int subarraysDivByK(vector& nums, int k) {
        unordered_map hash;
        hash[0 % k] = 1; // 0 这个数的余数

        int sum = 0,ret = 0;
        for(auto& n:nums)
        {
            sum += n;
            int r = (sum % k + k) % k; // 修正后的余数
            if(hash.count(r)) ret += hash[r];
            hash[r]++;
        }
        return ret;
    }
};

7、连续数组

(1) 题目解析

算法专题篇四:前缀和_第15张图片

    我们可以根据1,-1的性质,因为题目是要找到能够匹配相同数量的0,1数字。当我们将0替换为-1时,如果存在长度相同的0,1数对,那么它们的和为0。反过来,本题就是要找最长子数组元素和为0的长度。     

(2) 算法原理         

算法专题篇四:前缀和_第16张图片

class Solution {
public:
    int findMaxLength(vector& nums) {
        int n = nums.size();
        unordered_map hash;
        hash[0] = -1;
        int sum = 0;
        int len = 0;
        for(int i = 0; i < nums.size(); i++)
        {
            sum += nums[i] == 0 ? -1 : 1;   // 计算前缀和
            if(hash.count(sum)) len = max(len,i - hash[sum]);
            else hash[sum] = i; // 存储下标
        }
        return len;
    }
};

8、矩阵区域和

(1) 题目解析        

算法专题篇四:前缀和_第17张图片

        矩阵和同二维前缀和是相差无几的,都是求区域内的面积。例如上述例子,虽然是求整个矩阵的元素和,但是可以化解为从下标[1,1] 到 下标[3,3] 的前缀和矩阵。

(2) 算法原理

计算前缀和:

算法专题篇四:前缀和_第18张图片      

使用前缀和:

算法专题篇四:前缀和_第19张图片

class Solution {
public:
    vector> matrixBlockSum(vector>& mat, int k) {
        int m = mat.size();
        int n = mat[0].size();

        // 初始化前缀和
        vector> dp(m+1,vector(n+1));
        for(int i=1;i<=m;++i)
            for(int j=1;j<=n;++j) dp[i][j] = dp[i-1][j] + dp[i][j-1] + mat[i-1][j-1] - dp[i-1][j-1];

        // 使用前缀和
        vector> ret(m,vector(n));
        for(int i=0;i

本篇到此结束,感谢你的阅读。

祝你好运,向阳而生~

算法专题篇四:前缀和_第20张图片

你可能感兴趣的:(综合算法篇,算法,数据结构,c++)