Leetcode 面试题 17.24. 最大子矩阵 (最大子数组和的二维版本）

wwxy261

于 2020-08-23 17:48:00 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/108185977

1. 暴力做法枚举r1，c1, r2, c2，求和，时间复杂度O(N^6)很爆炸。

2. 优化方法，二维前缀和优化，O(1)时间内求出给定r1，c1，r2，c2区域的和，时间复杂度为O(N^4)

3. 降维动态规划优化。

枚举起始行和结束行，对列求和，等效成一维最大子序列问题。由于所有起始行和结束行都枚举过，所以这个做法一定是对的。

这道题目还需要记录更新r1，c1，r2，c2比一般的题目要难。

class Solution {
public:
    vector<int> getMaxMatrix(vector<vector<int>>& matrix) {
        int max = INT_MIN, n = matrix.size(), m = matrix[0].size();
        int r1, c1, r2, c2;
        for(int i=0;i<n;i++){
            vector<int> s(m);
            for(int j=i;j<n;j++){
                int cur = 0, tmp = 0;
                for(int k=0;k<m;k++){
                    s[k]+=matrix[j][k];
                    if(cur>0) cur+=s[k];