LeetCode--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?

題目解析：

方案一：不考虑空间限制

建立一个一模一样的二维数组O(mn)，其中也填充相同的数据。当在原数组中碰到0，就在备份数组中的i行j列相应的清除为0即可！这是最简单最直观的方案。

方案二：

如何进行优化？就跟n皇后问题一样，只需要用一行数组中某一个元素记录哪一行需要清零，用一列数组中的某一个元素记录哪一列需要清零。然后遍历这两个数组，对原数组清零即可。这样空间复杂度降低到O(m+n)。

方案三：

既然题目要求不适用额外空间，我们就从方案二找突破口。看能不能用某一列和某一行充当方案二的辅助空间。

---->思路一

先找到第一个为0的位置(i,j)，那么i行j列肯定要清零。这一行和这一列来当标记数组。当其他位置(k1,k2)为0时，就让(i,k2)和(k1,j)为0。扫描完以后，在根据这两个标记数组来进行相应的清空。

---->思路二：

既然我们要对某一行和某一列清零（i行j列），则相应的(i,0)和(0,j)肯定也要清零的。那么可以让0行和0列充当标记数组。

需要注意：

1：标记数组原始数据中是否已经存在0元素，要先做标记。然后忽略这两个数组存在的价值，专门去标记，最后再对两个数组操作。

2：第一遍的时候，只扫描，对标记数组进行清零，其他元素不操作。最后再统一清零。

class Solution {
public:
void setZeroes(vector<vector<int> > &matrix)   
  {  
      int m=matrix.size();  
      int n=matrix[0].size();  
      if(m==0||n==0)  
          return;  
     bool row=false;  
     bool col=false;  
        
    //验证第一列有没有0  
    for(int i=0;i<m;i++)  
    {  
      if(matrix[i][0]==0)  
          col=true;  
    }  
      //验证第一行有没有0  
      for(int j=0;j<n;j++)  
      {  
        if(matrix[0][j]==0)  
            row=true;  
      }  
        
      //标记数组  
      for(int i=1;i<m;i++)  
      {  
        for(int j=1;j<n;j++)  
        {  
           if(matrix[i][j]==0)  
           {//当某一元素为0，则将第0行，第j个元素置0；第i行，第0列个元素置0  
              matrix[i][0]=0;  
              matrix[0][j]=0;  
           }  
        }  
      }  
      //根据标记数组，将第i行置0  
      for(int i=1;i<m;i++)  
      {  
        if(matrix[i][0]==0)  
        {  
          for(int k=1;k<n;k++)  
              matrix[i][k]=0;  
        }  
            
      }  
        
      for(int j=1;j<n;j++)  
      {  
        if(matrix[0][j]==0)  
        {  
          for(int k=1;k<m;k++)  
              matrix[k][j]=0;  
        }  
      }  
      //如果原数组第1行有0  
      if(row)  
      {  
        for(int j=0;j<n;j++)  
            matrix[0][j]=0;  
      }  
      if(col)  
      {  
        for(int i=0;i<m;i++)  
            matrix[i][0]=0;  
      }  
       
  }  
};