《深入浅出进阶篇》——洛谷P1719 最大加权矩形 ——二维数组前缀和

上链接：https://www.luogu.com.cn/problem/P1719#submit

上题干：

题目描述

为了更好的备战 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

和为 15。

几个女孩子有点犯难了，于是就找到了电脑组精打细算的 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

这是一道二维数组的前缀和模板题，我们可以用画图来理解：

我们先从答案开始往前递推：

假设我们需要求任意一个子矩阵的矩阵和：设（x1，y1）为左上角，（x2，y2）为右下角

 

我们利用一维前缀和的思想：求（x，y）区间的字段和 为 S【y】-  S【x】

可以把这个小矩形，变成几个矩形的差值。

设S【x，y】为以【x，y】为右下角的矩阵和

那么则有

以（x1，y1）为左上角，（x2，y2）为右下角的矩阵和   =S【x2，y2】+S【x1-1，y1-1】-S【x2，y1-1】-S【x1-1，y2】

这是S【x2，y2】+S【x1-1，y1-1】（绿色部分为双重覆盖

这是 S【x2，y2】+S【x1-1，y1-1】-S【x2，y1-1】

这是S【x2，y2】+S【x1-1，y1-1】-S【x2，y1-1】-S【x1-1，y2】 

好了，现在我们要解决的就是怎么用S【x，y】表示【x，y】为右下角的矩阵和

我们还是用同样的思想，将S【x，y】分成多个矩阵和差

S【x，y】=S【x-1，y】+S【x-1，y】-S【x-1，y-1】+a【x，y】;

这是S【x-1，y】+S【x-1，y】  绿色部分为重叠部分

上代码： 

const int N = 130;
int a[N][N];
int s[N][N];
int sum = -0x7fffffff;
int qsum(int x1, int y1, int x2, int y2)
{
	return s[x2][y2] + s[x1 - 1][y1 - 1] - s[x2][y1 - 1] - s[x1 - 1][y2];
}
int main()
{
	int n;
	cin >> n;
	for(int i=1;i<=n;i++)
		for (int j = 1; j <= n; j++)
		{
			cin >> a[i][j];
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
		}
	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 = max(sum, qsum(x1, y1, x2, y2));
				}
	cout << sum;
}