前缀和

前缀和的作用是求一个数组中一段区间的和。

一维前缀和

一维前缀和的计算

for(int i = 1; i <= n; i++)
    pre_sum[i] = pre_sum[i-1] + nums[i];

ps：第一个元素从下标1开始，将0下标对应的元素置0，这样可以统一代码写法，不需要特判下标0了。

求一段区间[ l, r ] 之和

sum  = pre_sum[r] - pre_sum[l-1]

二维前缀和

二维前缀和的作用就是求一个子矩阵的元素之和。

二维前缀和数组的计算

ps：二维前缀和数组是指从左上角到当前位置[ i, j ] 的子矩阵之和。

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

ps：这里的下标也是从1开始，下标0对应的元素置空，理由同上

求[(x1, y1) , (x2, y2)] 子矩阵的元素之和

sum = pre_sum[x2][y2] - pre_sum[x2][y1-1] - pre_sum[x1-1][y2] + pre_sum[x1-1][y1-1]

 

前缀和的主要运用

前缀和可以和哈希表搭配

比如：给定一个字符串 str ，然后再给两个区间[l1, r1] [l2, r2]，判断这两个子串能否互相重新排列组成另一子串。（str由小写字母构成）

判断是否能组成另一子串是依据两子串中各字母是否数量相同。

可以使用一个数组s[26][n]，存储str的各字母出现的次数的前缀和。（每一行对应一个字母，可以看成26个一维数组，一维数组存的是在str中每个位置出现该字母的次数的前缀和）

然后根据所给的两个区间，可以计算出两个子串中各字母出现的次数，判断是否相同。

// 已知str是字符串，需要计算各字母出现的次数
// 
scanf("%s", str + 1); // 这里控制第一个字符为空
for(int i = 0; i < 26; i++)  // 循环26次，分别计算26个字母在字符串中出现次数的前缀和
    {
        for(int j = 1; str[j]; j++)
        {
            if(str[j] - 'a' == i)  // i 就是各字母的编号， 0 就是 a
                s[i][j] = s[i][j-1] + 1;
            else
                s[i][j] = s[i][j-1];
        }
    }


// 计算两个区间的字母出现次数是否相同
	for(int i = 0; i < 26; i++)
        {
        // 前面将前缀和数组第一个位置置0了，所以不需要特判
           if(s[i][r1] - s[i][l1-1] != s[i][r2] - s[i][l2- 1])
                {
                    flag = false; 
                    break;
                }
        }