Codeup100000632问题 E: 最大子矩阵

题目描述：

已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵，你的任务是找到最大的非空(大小至少是$1\times 1$)子矩阵。
比如，如下$4\times 4$的矩阵

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

的最大子矩阵是

9 2
-4 1
-1 8

这个子矩阵的大小是15。

输入：

输入是一个$N\times N$的矩阵。输入的第一行给出N ($0<N\le 100$)。
再后面的若干行中，依次（首先从左到右给出第一行的N个整数，再从左到右给出第二行的N个整数……）给出矩阵中的$N^2$个整数，整数之间由空白字符分隔（空格或者空行）。
已知矩阵中整数的范围都在[-127, 127]。

输出：

测试数据可能有多组，对于每组测试数据，输出最大子矩阵的大小。

样例输入：

1
27 
3
-40 29 -16 
38 18 22 
24 -35 5 

样例输出：

27
78

实现代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn = 100 + 10;
const int INF = 0x3fffffff;
int N;
int a[maxn][maxn];
int b[maxn][maxn];
int dp[maxn];
int mx;

//一维的动态规划求解连续最大子序和
void solve(int j) {
    memset(dp, 0, sizeof(dp));
    for(int i = 1; i <= N; i++) {
        dp[i] = max(b[j][i], dp[i - 1] + b[j][i]);
        mx = max(mx, dp[i]);
    }
}

int main() {
    while(scanf("%d", &N) != EOF) {
        mx = -INF;
        for(int i = 1; i <= N; i++) {
            for(int j = 1; j <= N; j++) {
                scanf("%d", &a[i][j]);
            }
        }
        // i表示起始行号，j表示结束行号，b[j][k]表示第k列从i到j行的元素和
        for(int i = 1; i <= N; i++) {
            memset(b, 0, sizeof(b));
            for(int j = i; j <= N; j++) {
                for(int k = 1; k <= N; k++) {
                    b[j][k] = a[j][k] + b[j - 1][k];
                }
                solve(j);
            }
        }
        printf("%d\n", mx);
    }
    return 0;
}