一维 + 二维前缀和详解

一维 + 二维前缀和详解

一、前缀和的作用

​ 算出一段区间内的和。常用求解的是一维区间或者二维区间

二、一维前缀和

1、 利用数组学会求解一维前缀和

例题1、

现在有N个的正整数放到数组里，现在要求一个新的数组 ，新数组的第i个数 是原数组第0到第i个数的和。

输入：

1 2	5 1 2 3 4 5

输出：

1	1 3 6 10 15

代码详解：

#include<iostream>
using namespace std;
const int maxl = 1e7 + 7;

int n, a[maxl], sum;//暴力模拟求解，时间复杂度O(n^2)

int main(){
	cin >> n;
	for (int i = 1; i <= n; i++)	cin >> a[i];
	for (int i = 1; i <= n; i++){
		sum = 0;
		for (int j = 1; j<= i; j++){
			sum += a[j];
		}
		cout << sum << " ";
	}
	return 0;
}

#include<iostream>
using namespace std;
const int maxl = 1e7 + 7;

int n, a, presum[maxl];//使用前缀和数组，算法时间复杂度O(n)

int main(){
	cin >> n;
	for (int i = 1; i <= n; i++){
		cin >> a;
		presum[i] = presum[i - 1] + a;
	}
	for (int i = 1; i <= n; i++)	cout << presum[i] << " ";
	return 0;
}

2、一维前缀和的应用，求解某一段区间和最值

例题2、

洛谷P1115 最大子段和

题目描述

给出一个长度为 n 的序列 a，选出其中连续且非空的一段使得这段和最大。

输入格式

第一行是一个整数，表示序列的长度 n。

第二行有 n 个整数，第 i个整数表示序列的第 i 个数字 a_i。

输出格式

输出一行一个整数表示答案。

输入输出样例

输入 #1复制

7
2 -4 3 -1 2 -4 3

输出 #1复制

4

说明/提示

样例 1 解释

选取 [3,5] 子段 {3,−1,2}{3,−1,2}，其和为 44。

数据规模与约定

• 对于 40%40% 的数据，保证 n≤2×10³。
• 对于 100%100% 的数据，保证 1≤n≤2×10⁵ ，−10⁴≤a_i≤10⁴ 。

代码详解：

#include<iostream>
using namespace std;
const int maxl = 1e4 + 7;

int n, s = 0, ans = -2e9, temp;

int main(){
	cin >> n;
	for (int i = 0; i < n; i++){
		cin >> temp;
		if (temp + s >= temp)	s += temp;//比我当前的值还要小，就说明这一区间已经不是最大的了
		else	s = temp;
		ans = max(ans, s);
	}
	cout << ans << endl;
	return 0;
}

三、二维前缀和

例题1、

洛谷P1719 最大加权矩形

题目描述

为了更好的备战 NOIP2013，电脑组的几个女孩子 LYQ,ZSC,ZHQ 认为，我们不光需要机房，我们还需要运动，于是就决定找校长申请一块电脑组的课余运动场地，听说她们都是电脑组的高手，校长没有马上答应他们，而是先给她们出了一道数学题，并且告诉她们：你们能获得的运动场地的面积就是你们能找到的这个最大的数字。

校长先给他们一个 n×n 矩阵。要求矩阵中最大加权矩形，即矩阵的每一个元素都有一权值，权值定义在整数集上。从中找一矩形，矩形大小无限制，是其中包含的所有元素的和最大 。矩阵的每个元素属于 [−127,127] ,例如

 0 –2 –7  0 
 9  2 –6  2
-4  1 –4  1 
-1  8  0 –2

在左下角：

9  2
-4  1
-1  8

和为 1515。

几个女孩子有点犯难了，于是就找到了电脑组精打细算的 HZH，TZY 小朋友帮忙计算，但是遗憾的是他们的答案都不一样，涉及土地的事情我们可不能含糊，你能帮忙计算出校长所给的矩形中加权和最大的矩形吗？

输入格式

第一行：n，接下来是 n行 n 列的矩阵。

输出格式

最大矩形（子矩阵）的和。

输入输出样例

输入 #1复制

4
0 -2 -7 0
 9 2 -6 2
-4 1 -4  1 
-1 8  0 -2

输出 #1复制

15

说明/提示

1≤n≤120

求解思路：

二维前缀和求解思路

代码详解：

#include<iostream>
using namespace std;
const int maxl = 200;

int n, presum[maxl][maxl], a[maxl][maxl], sumMax, sum;

int main(){
	cin >> n; 
	for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)	cin >> a[i][j];
			
	for (int i = 1; i <= n; i++){
		for (int j = 1; j <= n; j++){
			presum[i][j] = presum[i - 1][j] + presum[i][j - 1] - presum[i - 1][j - 1] + a[i][j];
		}
	}
	
	sumMax = a[1][1];
	for (int x1 = 1; x1 <= n; x1++){
		for (int y1 = 1; y1 <= n; y1++){
			for (int x2 = x1 ; x2 <= n; x2++){
				for (int y2 = y1; y2 <= n; y2++){
					sum = presum[x2][y2] - presum[x1 - 1][y2] - presum[x2][y1 - 1] + presum[x1 - 1][y1 - 1];
					sumMax = max(sumMax, sum);
				}
			}
		}
	}
	cout << sumMax << endl;
	return 0;
}