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：第一遍的时候，只扫描，对标记数组进行清零，其他元素不操作。最后再统一清零。

  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;
        }
       
    }