最大子矩阵+01矩阵

目录

最大子矩阵

题目

思路

代码

01矩阵

题目

思路

代码

---

最大子矩阵

题目

思路

确定一个矩阵中的子矩阵，只需要确定矩阵对角线上两点的坐标即可，

在确定了子矩阵之后，还需要计算子矩阵中各元素之和，如果按照常规的循环方法来解决，时间复杂度是O(N^6)，对于本题200长度大小，会超时，所以应该再另寻方法，仔细思考一番发现，确定子矩阵的四层循环，似乎是在所难免的，那么接下来的重点就是处理子矩阵中各元素和上边了，可以采用如下方法：定义一个同原始矩阵大小一样的矩阵，定义dp[i][j]为以[0][0]和[i][j]顶点矩阵的各元素之和，如果提前计算好，即将dp[][]的初始化和填表完成在寻找子矩阵之前，那么时间复杂度就变成了O(N^4)，当确定好dp[][]之后，当确定了子矩阵的对角线上两点的坐标后，例如（x1,y1）和（x2,y2），将采取如下方法进行计算

填写dp[][]表时，当填写红色位置的值时，只需将S(紫)+S(绿)-S(蓝)+红色位置在原始矩阵中对应的值即可，即dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+matrix[i-1][j-1].(假设dp表比原始矩阵多一行，多一列)。

那么接下来在完成dp表的填写后，确定子矩阵的元素之和大小可采用如下方法：

确定红色区域之和=整个区域-绿色区域-紫色区域+橙色区域

即S(红）= dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1].

在完成以上点之后，接下来确定最大子矩阵的坐标就信手拈来了，定义一个变量记录之前的最大值，当遇到破纪录的值时，更新最大子矩阵的坐标。

代码

class Solution {
public:
    int dp[210][210];

    vector<int> getMaxMatrix(vector<vector<int>>& matrix) {
        vector<int> ans(4);
        int a,b,c,d;
        int m=matrix.size(),n=matrix[0].size();
        for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
                dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+matrix[i-1][j-1];
        int ma=INT_MIN;
        for(int x1=1;x1<=m;x1++)
            for(int y1=1;y1<=n;y1++)
                for(int x2=x1;x2<=m;x2++)
                    for(int y2=y1;y2<=n;y2++)
                    {
                        int ret=dp[x2][y2]-dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1];
                        if(ret>ma){
                            ma=ret;
                            // ans[0]=x1-1,ans[1]=y1-1,ans[2]=x2-1,ans[3]=y2-1;
                            a=x1-1,b=y1-1,c=x2-1,d=y2-1;
                        }
                    }
        ans[0]=a,ans[1]=b,ans[2]=c,ans[3]=d;
        return ans;
    }
};

经过试验发现，如果采用注释掉的方法进行更新，会超时，但是采用上边的方法，就可以通过了。

01矩阵

题目

思路

采用广度优先遍历来解决，创建一个同原始矩阵同样大小的矩阵dists，矩阵中各位置的值为该点距离0的最短距离，对于原始矩阵中值为0的点，在dists矩阵中对应的值为0，首先将原始矩阵中值为0的点放到队列中，然后不断地进行广度优先遍历，对于没遍历过的点，放入队列中，直到队列为空，更新结束。

代码

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
        int m=mat.size(),n=mat[0].size();
        vector<vector<int>> dists(m,vector<int>(n,INT_MAX));
        queue<pair<int,int>> q;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
            {
                if(mat[i][j]==0){
                    q.push({i,j});
                    dists[i][j]=0;
                }   
            }
        int dx[4]={0,0,1,-1};
        int dy[4]={-1,1,0,0};
        while(!q.empty()){
            int a=q.front().first;
            int b=q.front().second;
            int dist=dists[a][b];
            q.pop();
            for(int k=0;k<4;k++){
                int x=a+dx[k];
                int y=b+dy[k];
                if(x>=0 && x<m && y>=0 && y<n){
                    if(dists[x][y]>dist+1){
                        dists[x][y]=dist+1;
                        q.push({x,y});
                    }
                }
            }
        }
        return dists;
    }
};