Acwing-126. 最大的和（二维前缀和）

126. 最大的和

给定一个包含整数的二维矩阵，子矩形是位于整个阵列内的任何大小为1 * 1或更大的连续子阵列。

矩形的总和是该矩形中所有元素的总和。

在这个问题中，具有最大和的子矩形被称为最大子矩形。

例如，下列数组：

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

其最大子矩形为：

9 2 
-4 1 
-1 8 

它拥有最大和15。

输入格式

输入中将包含一个N*N的整数数组。

第一行只输入一个整数N，表示方形二维数组的大小。

从第二行开始，输入由空格和换行符隔开的$N^2$个整数，它们即为二维数组中的$N^2$个元素，输入顺序从二维数组的第一行开始向下逐行输入，同一行数据从左向右逐个输入。

数组中的数字会保持在[-127,127]的范围内。

输出格式

输出一个整数，代表最大子矩形的总和。

数据范围

1≤N≤1001≤N≤100

输入样例：

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

输出样例：

15

难度：简单
时/空限制：1s / 64MB
总通过数：168
总尝试数：293
来源：《算法竞赛进阶指南》

链接：

https://www.acwing.com/problem/content/128/

思路：

方法一：
我们可以使用二维前缀和来做，我们需要遍历子矩形的左上点（i, j）和右下点(k,p)，通过二维前缀和来求出矩形中元素的和。

我们可以推出$F[i][j]=F[i-1][j]+F[i][j-1]+A[i][j]-F[i-1][j-1]$
当我们求左上点为(i,j)，右下点为(k,p)的矩阵和时，我们可以得出
$val=F[k][p]-F[k][j]-F[i][p]+F[i][j]$
直接暴力找两个点，复杂度$O(n^4)$

代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
int a[maxn][maxn];
int sum[maxn][maxn];
const int INF = -0x3f3f3f3f;
int main()
{
    ios::sync_with_stdio(false);
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            cin >> a[i][j];

            sum[i][j] = a[i][j]+ sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
        }
    }
    int maxx = INF;
    for(int i = 0; i <= n; i++) {
        for(int j = 0; j <= n; j++) {
            for(int k = i; k <= n; k++) {
                for(int p = j; p <= n; p++) {
                    int ans = sum[k][p] - sum[k][j] - sum[i][p] + sum[i][j];
                    if(ans > maxx)
                        maxx = ans;

                }
            }
        }
    }
    cout << maxx << endl;
    return 0;
}

方法二：
我们可以先用求一维最大连续子序列的和来求。二维我们可以把几层的高度和为一层来算。这样我们就变为了一维。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 109;
int a[maxn][maxn];
int sum[maxn][maxn];
int main()
{
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            cin >> a[i][j];
            a[i][j] += a[i-1][j];
        }
    }
    int maxx = -0x3f3f3f3f;
    for(int i = 1; i <= n; i++) {
        for(int j = i; j <= n; j++) {
            int last = 0;
            for(int k = 1; k <= n; k++) {
                last = max(0, last);
                int ans = a[j][k] - a[i-1][k];
                maxx = max(maxx,ans+last);
                last += ans;
            }
        }
    }
    cout << maxx << endl;
    return 0;
}