2020-5 leetcode 面试题 17.24. 最大子矩阵

1.设计一个二维数组,每个元素是一个二元组,第一个元素存储从p 列开始,到 q 列结束,的这几列。从第一行开始加到现在所在行的和。第二个元素 存储 和的值最新一次大于等于0 所在的行。
2.总的来说 是用一个二维组,存储列的排列情况。其存储的值,使得循环重数减少到三重。(少了一层关于行的循环)


class Solution {
public:
    vector<int> getMaxMatrix(vector<vector<int>>& matrix) {
        int n=matrix.size(),m=matrix[0].size();
        vector<vector<vector<int>>> dp(m,vector<vector<int>>(m,vector<int>(2,0)));
        int ans=INT_MIN;
        vector<int> ans1(4,-1);
        
        for(int i=0;i<n;i++){
            vector<int> row_sum(m+1,0);
            row_sum[1]=matrix[i][0];
            for(int j=2;j<=m;j++){
                row_sum[j]=row_sum[j-1]+matrix[i][j-1];
               
            } 


            for(int p=0;p<m;p++){
                for(int q=p;q<m;q++){
                    dp[p][q][0] +=row_sum[q+1]-row_sum[p];
                   
                    if(dp[p][q][0]>ans){
                        ans1[0]=dp[p][q][1];
                        ans1[1]=p;
                        ans1[2]=i;
                        ans1[3]=q;
                        ans=dp[p][q][0];
                    }
                    if(dp[p][q][0]<=0){
                        dp[p][q]={0,i+1};
                    }
                }
            }
        }
        return ans1;
    }
};

你可能感兴趣的:(leetcode)