leetcode - 73. Set Matrix Zeroes

    xiaoxiao2021-03-26  30

    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.

    click to show follow up.

    Follow up:

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

    Subscribe to see which companiesasked this question.

     

     

    这一题到最后也只想出了利用O(m+n)的空间来进行求解,就是分别用n和m个长度的数组来记录标识哪一行和哪一列要赋值为0。

     

    然后看了网上别人的代码,发现真的非常巧妙,利用已经找到的0的那一行和那一列来作为额外的开销行和开销列,最后不用额外的空就就能进行求解。

     

     

     

     

     

     

     

     

    算法思想:

    先遍历一遍二维数组,找到1个0,然后记录下这个0所在的行和列的序号,然后利用找到的行和列来作为一个记录数组,记录哪些行哪些列需要赋值为0,最后只需要遍历一次这两个记录数组就可以对原始数组进行赋值,求得解。

     

     

     

     

     

    步骤:

    1.      遍历1次数组,与0进行比较:

    a)        若与0相等,则停止遍历,记录下当前位置的行号x,列号y

    b)        若都不与0相等,则直接返回矩阵(说明矩阵中没有0存在)

    2.      初始化第x行和第y列为-1;

    3.      遍历原始数组,记当前遍历元素为a[i][j],判断当前元素:

    a)        若为0,则a[x][j] = 1 , a[i][y] = 1,表示在x行和y列记录下要赋值为0的行号和列号。

    4.      遍历x行和y列,对等于1的行和列进行赋值为0;

    class Solution { public: void setZeroes(vector<vector<int>>& a) { if(a.size() < 1)return ; int x,y; int find_zero = 0; x = -1 ; y = -1; for(int i = 0 ; i < a.size() ; i++){ if(find_zero)break; for(int j = 0 ; j < a[0].size() ; j++){ if(a[i][j])continue; find_zero = 1; x = i ; y = j; break; }//for j }//for i if(!find_zero) return; for(int i = 0 ; i < a.size() ; i++)if(a[i][y] != 0) a[i][y] = -1; for(int i = 0 ; i < a[0].size() ; i++)if(a[x][i] != 0) a[x][i] = -1; for(int i = 0 ; i < a.size() ; i++){ for(int j = 0 ; j < a[0].size() ; j++){ if(!a[i][j]){ a[x][j] = 1; a[i][y] = 1; }// end if }//for j }//for i for(int i = 0 ; i < a.size() ; i++) if(a[i][y] == 1 && i != x)a[i].assign( a[0].size(),0); for(int i = 0 ; i < a[0].size() ; i++) if(a[x][i] == 1) for(int k = 0 ; k < a.size() ; k++) a[k][i] = 0; for(int i = 0 ; i < a.size() ; i++)a[i][y] = 0; for(int i = 0 ; i < a[0].size() ; i++)a[x][i] = 0; } };

     

     

    转载请注明原文地址: https://ju.6miu.com/read-664375.html

    最新回复(0)