子矩阵的最大累加和问题

https://www.nowcoder.com/practice/cb82a97dcd0d48a7b1f4ee917e2c0409tpId=101&tqId=33095&tPage=1&rp=1&ru=/ta/programmer-code-interview-guide&qru=/ta/programmer-code-interview-guide/question-ranking

题目描述

给定一个矩阵matrix，其中的值有正、有负、有0，返回子矩阵的最大累加和

例如，矩阵matrix为：

-90 48 78

64 -40 64

-81 - 7 66

其中，最大的累加和的子矩阵为：

48 78

-40 64

-7 66

所以返回累加和209。

例如，matrix为：

-1 -1 -1

-1 2 2

-1 -1 -1

其中，最大累加和的子矩阵为：

2 2

所以返回4

[要求]

时间复杂度为O(n^2m)O(n2m)，空间复杂度为O(nm)O(nm)

 

输入描述:

第一行有两个整数N，M。分别表示矩阵的行数/列数
接下来N行，每行M个整数表示矩阵内的数

输出描述:

输出一个整数表示答案

示例1

输入

3 3
-90 48 78
64 -40 64
-81 -7 66

输出

209

示例2

输入

3 3
-1 -1 -1
-1 2 2
-1 -1 -1

输出

4

备注:

1⩽N,M⩽200

−100⩽arri,j​⩽100

如果矩阵中没有正数，返回矩阵中的最大值。

 

解决这个问题前先来个铺垫

https://www.nowcoder.com/practice/38014783330445b4936ccfad9990431ctpId=101&tqId=33094&tPage=1&rp=1&ru=/ta/programmer-code-interview-guide&qru=/ta/programmer-code-interview-guide/question-ranking

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int> arr(n);
    for(int i=0;i<n;i++){
        cin>>arr[i];
    }
    int res=INT_MIN;
    int cur=0;
    for(int i=0;i<n;i++){
        res=max(res,cur+arr[i]);
        cur=max(cur+arr[i],0); 
    }
    cout<<res;
    return 0;
}

然后将一维原理扩展到二维平面，注意是子矩阵的最大累积和问题，不忘简单理解成最大值问题

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    vector<vector<int>> matrix(n,vector<int>(m,0));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>matrix[i][j];
        }
    }
    int res=INT_MIN;
    int cur=0;
    for(int i=0;i<n;i++){
        vector<int> tmp(m);
        for(int j=i;j<n;j++){
            cur=0;
            for(int k=0;k<m;k++){
                tmp[k]+=matrix[j][k];
                cur+=tmp[k];
                res=max(res,cur);
                cur=max(cur,0);
            }
        }
    }
    cout<<res<<endl;
    return 0;
}