LeetCode 73. Set Matrix Zeroes

LeetCode 73. Set Matrix Zeroes


题目描述:

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

Example 1:

Input: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
Output: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]
Example 2:

Input: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
Output: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

Follow up:

  • A straight forward solution using O(mn) space is probably a bad idea.
  • A simple improvement uses O(m + n) space, but still not the best solution.
  • Could you devise a constant space solution?

题目理解:

将给定矩阵中0元素所在的行列都变成0.提示里面也说了,要求降低空间复杂度,暴力的直接申请一个M*N的空间就很舒服,或者申请两个vector记录为0的行和列这样复杂度就降低成了M+N,然后借鉴了网上优秀代码将空间复杂度降低成了1.


暴力美学:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();
        int line = matrix[0].size();
        vector<vector<int>> ans(matrix);
        for(int i=0;i<row;i++){
            for(int j=0;j<line;j++){
                if(matrix[i][j] == 0){
                    for(int p=0;p<row;p++) ans[p][j] = 0;
                    for(int q=0;q<line;q++) ans[i][q] = 0;
                }
            }
        }
        matrix.swap(ans);
    }
};

学习优秀代码:

思路理解:
- 在数组的第一列和第一行先找看看是否存在0,并将其标志存储起来;
- 然后遍历剩余数组,若发现0,则将对应位置的第一行和第一列元素变成0,这样因为之前第一行列是否有零的标志已经被记录了下来,所以这里不用担心破坏;
- 在找完0后,对除了第一行列外的剩余矩阵进行操作,用对应行列在第一行列的元素是否为0判断该行列是否都置0;
- 都完成后,再根据最开始存储的标志对第一行列,进行变0操作。


class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();
        int line = matrix[0].size();
        bool rflag,lflag;
        rflag = lflag = false;
        for(int i=0;i<row;i++){
            if(!matrix[i][0]){//遍历第一列
                lflag = true;
            } 
        }
        for(int j=0;j<line;j++){
            if(!matrix[0][j]){
                rflag = true;
            }
        }
        for(int i=1;i<row;i++){
            for(int j=1;j<line;j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for(int i=1;i<row;i++){
            for(int j=1;j<line;j++){
                if(matrix[i][0] == 0 || matrix[0][j]==0){
                    matrix[i][j] = 0;
                }
            }
        }
        if(lflag){
        for(int i=0;i<row;i++){
                matrix[i][0] = 0;
            }
        }
        if(rflag){
        for(int j=0;j<line;j++){
                matrix[0][j] = 0;
            }
        }
    }
};

空间复杂度可以降低,时间复杂度无法操作。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值