162.Set Matrix Zeroes-矩阵归零（中等题）

矩阵归零

1. 题目

   给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。
   需要在原矩阵上完成操作。

2. 样例

   给出一个矩阵
   
   返回
   

3. 挑战

   你是否使用了额外的空间？
   一个直接的解决方案是使用 O(MN) 的额外空间，但这并不是一个好的方案。
   一个简单的改进方案是使用 O(M + N) 的额外空间，但这仍然不是最好的解决方案。
   你能想出一个常数空间的解决方案吗？

4. 题解

   1.先扫描一遍第一行和第一列，看是否有0，并进行标记。
   2.从第二行和第二列开始对矩阵进行遍历，如matrix[i][j]为0，则将matrix[0][j]和matrix[i][0]都设为0。
   3.再以2同样的方式进行遍历，对任意matrix[i][j]，如果其对应的第1行或第1列的元素为0，则将matrix[i][j]也设置为0。
   4.如果步骤1记录的第一行和第一列有0，则将该行列全部置为0。

public class Solution {    /**     * @param matrix: A list of lists of integers     * @return: Void     */    public void setZeroes(int[][] matrix) {        if(matrix == null || matrix.length == 0)        {            return;        }        int rows = matrix.length;        int cols = matrix[0].length;        boolean firstRowHasZero = false;        boolean firstColHasZero = false;        for (int i=0;i<cols;i++)        {            if (matrix[0][i] == 0)            {                firstRowHasZero = true;                break;            }        }        for (int i=0;i<rows;i++)        {            if (matrix[i][0] == 0)            {                firstColHasZero = true;                break;            }        }        for(int i = 1; i < rows; i++)         {            for(int j =1; j<cols; j++)            {                if(matrix[i][j] == 0)                {                    matrix[0][j] = 0;                    matrix[i][0] = 0;                }            }        }        for(int i = 1; i<rows; i++)         {            for (int j=1; j< cols; j++)             {                if(matrix[0][j] == 0 || matrix[i][0] == 0)                {                    matrix[i][j] = 0;                }            }        }        if(firstRowHasZero)        {            for(int i = 0; i < cols; i++)            {                matrix[0][i] = 0;            }                   }        if(firstColHasZero)        {            for(int i = 0; i < rows; i++)            {                matrix[i][0] = 0;            }                   }    }}

Last Update 2016.10.18