Set Matrix Zeroes

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 use extra space?
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?

方法1: 利用O(mn)的额外空间

class Solution {
public:
    void setZeroes(vector
   
   
    
    
     
     >& matrix) {
        int m=matrix.size();
        vector
     
     
      
      
       
       > index;
        if(m==0)
        return;
        int n=matrix[0].size();
        for(int i=0;i
       
       
         t=index[index.size()-1]; index.pop_back(); for(int k=0;k 
         
       
      
      
     
     
    
    
   
   

方法2：用O(1)的额外空间

思想：用矩阵的第一行和第一列存储信息，判断该行或该列是否有0元素。因为matrix[0][0]既能代表第一行，也能代表第一列，所以这里另设一元素int col0来存储第一列是否有0元素，matrix[0][0]仅表示第一行，这样避免歧义。注意，第一次遍历，从高到低，第二次遍历从低到高，因为存储信息的第一行一定要放在最后。

class Solution {
public:
    void setZeroes(vector
    
    
     
     
      
      >& matrix) {
        int m=matrix.size();//m是行数
        if(m==0)
        return;
        int n=matrix[0].size();//n是列数
        int col0=1;//用来表示第一列的状态，如果有一个元素为0，设为0
        for(int i=0;i
      
      
       
       =0;i--)//好奇怪，为什么调换一下，就不对了？//for(i==0;i
       
       
        
        0;j--)
            if(matrix[i][0]==0||matrix[0][j]==0)
            matrix[i][j]=0;
            if(col0==0) matrix[i][0]=0;
        }
        
        
    }
};
       
       
      
      
     
     
    
    

第一次循环：

第二次循环：

假如还是从第一行开始遍历，如果matrix[0][0]=0,然后后面的信息会都变成0，结果就都是0了。