腳本之家服務器常用軟件

快捷導航

C++中前綴和數組(算法)基本介紹

更新時間：2024年12月13日 11:54:07 作者：貓咪-9527

前綴和（Prefix Sum）是指數組中某個位置之前的所有元素的和,本文將介紹C++中前綴和數組(算法)基本概念,感興趣的朋友一起看看吧

1.前言

如何快速求得一個數組的前綴和？

1. 1 前綴和的基本概念

前綴和（Prefix Sum）是指數組中某個位置之前的所有元素的和。對于數組arr，其前綴和數組prefixSum定義為：

prefixSum[0] = 0prefixSum[i] = prefixSum[i-1] + arr[i-1]（對于i > 0）
這樣，任意子數組arr[l...r]的和可以通過prefixSum[r+1] - prefixSum[l]快速計算出來。

2.一維數組的前綴和

在處理數組區(qū)間和問題時，前綴和（Prefix Sum）是一個非常有效的工具，可以大大加快查詢速度。下面詳細解釋如何預處理前綴和數組，并使用前綴和數組快速計算任意區(qū)間的元素和。

步驟一：預處理前綴和數組

給定一個數組arr，長度為n，我們可以預處理一個前綴和數組dp，其中dp[i]表示數組arr從索引1到索引i的所有元素的和。

初始化dp[0] = 0（這是為了方便計算從索引1開始的區(qū)間和）。

使用遞推公式dp[i] = dp[i - 1] + arr[i - 1]來計算dp數組的每個元素。注意，這里arr的索引從0開始，而dp的索引

從1開始模擬區(qū)間[1, i]。//從1開始是為了避免從0開始時會出現-1，從而進行判斷，減少復雜邊界情況

1 2 3 4 5

0

1

2

3

4

5

!!!此步是為了處理邊界情況

具體代碼如下：

std::vector<int> dpsum(const vector<int>& arr) {
    int n = arr.size();
    vector<int> dp(n + 1, 0); // dp 數組的長度是 n+1，并初始化為 0
    for (int i = 1; i <= n; ++i) {
        dp[i] = dp[i - 1] + arr[i - 1];
    }
    return dp;
}

步驟二：使用前綴和數組快速計算區(qū)間和

給定一個區(qū)間[l, r]，我們可以利用預處理好的前綴和數組dp快速計算區(qū)間內所有元素的和。

區(qū)間[l, r]內所有元素的和為dp[r] - dp[l - 1]。

具體解釋如下：

dp[r]包含從索引1到索引r的所有元素的和。
dp[l - 1]包含從索引1到索引l - 1的所有元素的和。
因此，dp[r] - dp[l - 1]正好是區(qū)間[l, r]內所有元素的和。

具體代碼如下：

int queryRangeSum(const vector<int>& dp, int l, int r) {
    return dp[r] - dp[l - 1];
}

下面展示一個一維數組前綴和模板

#include <iostream>
#include <vector>
using namespace std;
// 預處理前綴和數組的函數
vector<int> dpSum(const vector<int>& arr) {
    int n = arr.size();
    std::vector<int> dp(n + 1, 0); // dp 數組的長度是 n+1，并初始化為 0
    for (int i = 1; i <= n; ++i) {
        dp[i] = dp[i - 1] + arr[i - 1];
    }
    return dp;
}
// 查詢區(qū)間和的函數
int queryRangeSum(const vector<int>& dp, int l, int r) {
    return dp[r] - dp[l - 1];
}
int main() {
    // 示例數組
    vector<int> arr = {1, 2, 3, 4, 5};
    // 預處理前綴和數組
    vector<int> dp = dpSum(arr);
    // 查詢區(qū)間 [2, 4] 的和（注意C++中數組索引從0開始，但這里的l,r是區(qū)間表示，從1開始考慮）
    int sum = queryRangeSum(dp, 2, 4);
    cout << "Sum of range [2, 4]: " << sum << endl; // 輸出 9 (2+3+4)
    return 0;
}

3.二維數組求前綴和

二維數組求前綴和和一維數組求前綴和思路相似，都是通過預處理前綴和來解決此類問題。

類?于?維數組的形式，如果我們能處理出來從 [0, 0] 位置到 [i, j] 位置這?區(qū)域內所有元素的累加和，就可以在 O(1) 的時間內，搞定矩陣內任意區(qū)域內所有元素的累加和。因此我們接下來僅需完成兩步即可：

1	2	3
4	5	6
7	8	9

0	0	0
1	2	3
4	5	6
7	8	9

for(int i = 1; i <= n; i++) 
 for(int j = 1; j <= m; j++)
 cin >> arr[i][j];

這樣，我們填寫前綴和矩陣數組的時候，下標直接從 1 開始，能?膽使? i - 1 , j - 1 位置的值。

此時我們所求的[i,j]坐標的和，即為下圖藍色區(qū)域位置

 // 處理前綴和矩陣
 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];

將表格抽象為四部分：分別為綠，粉，藍，白

	x		i
	0	0	0	0
j	0	1	2	3
	0	4	5	6
y	0	7	8	9

如果我們想求出白色部分的和

即為：總-綠-粉-藍=白；

綠+粉=dp[i][j];

綠+藍=dp[x][y];

白=dp[i][y]-dp[x][y]-dp[i][j]+dp[x][j];

總代碼如下

#include <iostream>
using namespace std;
const int N = 1010;
int arr[N][N];
long long dp[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++)
 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;
 }
}

4.二維矩陣中心前綴和

二維矩陣中心前綴和求取方式和二維矩陣前綴和方法類似，不過需要多進行處理幾步。

中心坐標[x,y]的寬k=1中心前綴和，為以下藍色部分

x
y	1	2	3
	4	5	6
	7	8	9

坐標[x,y]的中心前綴和，為以下藍色部分

x		1	2	3
y	4	5	6
	7	8	9

此時步驟為：

1.先求普通二維前綴和dp，建立首行列都為0的普通前綴和數組

	x		i
	0	0	0	0
j	0	1	3	6
	0	5	12	21
y	0	12	27	55

2.求建立新的中心前綴和數組即為(去掉為0的輔助項)：

x1=max(0,i-k);y1=max(0,j-k);//避免出現越界情況

arr[i][j]=dp[x1-1][y1-1]+dp[x2][y2]-dp[x1-1][y2]-dp[x2][y2]

以下是完整代碼：

class Solution {
public:
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) {
        int m=mat.size(),n=mat[0].size();
        vector<vector<int>> dp(m+1,vector<int>(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]-dp[i-1][j-1]+mat[i-1][j-1];
            }
        }
         vector<vector<int>> arr(m,vector<int>(n));
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                int x1=max(i-k,0)+1;
                int y1=max(j-k,0)+1;
                int x2=min(i+k,m-1)+1;
                int y2=min(j+k,n-1)+1;
               arr[i][j]=dp[x1-1][y1-1]+dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]; 
            }
        }
        return arr;
    }
};

5.前綴和與哈希表相結合

前綴和與哈希表相結合是一種非常有效的算法技巧，常用于解決數組或字符串中與子數組（或子串）和相關的問題。這種方法的核心思想是通過前綴和來快速計算任意子數組的和，并利用哈希表來記錄這些和出現的頻率，從而高效地解決問題。

1. 哈希表的作用

哈希表（Hash Table）用于記錄前綴和出現的頻率。在處理問題時，我們可以利用哈希表快速查找某個前綴和是否已經出現過，以及出現了多少次。

(假設需要求出i前面的等于k的子數組個數)

0x1x2……i

對于i來說數組可以分為兩段

ki的前綴和-ki

2. 前綴和與哈希表相結合的應用

假設我們需要求出數組中所有和為k的子數組的個數。我們可以按照以下步驟進行：

初始化一個哈希表count，用于記錄前綴和出現的頻率。將count[0]初始化為1，表示前綴和為0的情況（即空子數組）出現了1次。
初始化一個變量result為0，用于記錄和為k的子數組的個數。
遍歷數組arr，計算當前位置的前綴和prefixSum[i]。
計算目標前綴和target = prefixSum[i] - k。如果target在哈希表中出現過，說明存在一個或多個子數組的和為k（這些子數組的右端點是當前位置i，左端點可以通過哈希表中的記錄確定）。
將result增加count[target]的值。
更新哈希表count，將prefixSum[i]的頻率增加1。
遍歷結束后，result就是和為k的子數組的個數。

#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;
int numSubarraySumEqualK(vector<int>& nums, int k) {
    unordered_map<int, int> count; // 哈希表，記錄前綴和出現的頻率
    count[0] = 1; // 初始化前綴和為0的頻率為1（表示空子數組）
    int prefixSum = 0; // 當前位置的前綴和
    int result = 0; // 和為k的子數組個數
    for (int num : nums) {
        prefixSum += num; // 計算當前位置的前綴和
        int target = prefixSum - k; // 計算目標前綴和
        if (count.find(target) != count.end()) {
            // 如果目標前綴和在哈希表中出現過，則增加結果
            result += count[target];
        }
        // 更新哈希表中當前前綴和的頻率
        count[prefixSum]++;
    }
    return result;
}
int main() {
    vector<int> nums = {1, 1, 1}; // 示例數組
    int k = 2; // 目標和
    int result = numSubarraySumEqualK(nums, k);
    cout << "和為" << k << "的子數組個數為: " << result << endl;
    return 0;
}

到此這篇關于C++中前綴和數組(算法)基本介紹的文章就介紹到這了,更多相關c++前綴和數組內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: