牛客网 最大子矩阵（dp，求子矩阵最大和）

题目传送门

最大子矩阵

题目描述

已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵，你的任务是找到最大的非空(大小至少是1 * 1)子矩阵。 比如，如下4 * 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 * N的矩阵。输入的第一行给出N (0 < N <= 100)。
再后面的若干行中，依次（首先从左到右给出第一行的N个整数，再从左到右给出第二行的N个整数……）给出矩阵中的N2个整数，整数之间由空白字符分隔（空格或者空行）。
已知矩阵中整数的范围都在[-127, 127]。

输出描述:

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

输入

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

输出

15

思路

转化为最大子序列和计算，枚举遍历起始行 i 和终止行 j，$i=j$，寻找原矩阵的该行的最大子序列和，$i\ne j$ 的话 将 i , i+1 , … , j 的各行元素加到一行上，然后计算这一行的最大子序列和。

构造一个矩阵total，便于计算每行的叠加和，$total[i][j]={\sum }_{x=0}^{i}total[x][j]$

help为转化的一维数组

代码

#include <bits/stdc++.h>
#define MAXN 105
using namespace std;
int arr[MAXN][MAXN];
int total[MAXN][MAXN];
int help[MAXN];
int dp[MAXN];
int n;
void init(){
    memset(arr,0,MAXN);
    memset(total,0,MAXN);
    memset(help,0,MAXN);
    memset(dp,0,MAXN);
}

int findMax(){
    dp[0] = help[0];
    int ans = dp[0];
    for (int i=1;i<n;i++){
        dp[i]=max(help[i],dp[i-1]+help[i]);
        ans=max(dp[i],ans);
    }
    return ans;
}

int getArr(){
    int ans = -1;
    for (int i=0;i<n;i++){
        for (int j=i;j<n;j++){
            for (int k=0;k<n;k++){
                if (i==0){
                    help[k]=total[j][k];
                }else {
                    help[k]=total[j][k]-total[i-1][k];
                }
            }
            ans = max(ans,findMax());
        }
    }
    return ans;
}


int main (){
    while (scanf("%d",&n)!=EOF){
        init();
        for (int i=0;i<n;i++){
            for (int j=0;j<n;j++){
                scanf("%d",&arr[i][j]);
                total[i][j]=arr[i][j];
                if (i>0){
                    total[i][j]+=total[i-1][j];
                }
            }

        }
        int ans = getArr();
        printf("%d\n",ans);

    }
    return 0;
}