LeetCode-73.矩阵置零（相关话题：数组）

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 1:

输入: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2:

输入: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

进阶:

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

1.O(m + n)的额外空间

解题思路：boolean[] line和boolean[] col，line[i]记录第i行是否有0，col[j]记录第j列是否有0.

遍历数组matrix[i][j]，
1.若matrix[i][j] != 0，判断line[i]或col[j]是否为true，若为true，则matrix[i][j] = 0；
2.若matrix[i][j] == 0，
    2.1判断line[i]，如果为false，则将matrix[i][0~j-1]置为0(注：不是将第i行的所有元素置0，若将所有元素置0，可能会将原本不必置0的列置0)，并且line[i] = true
    2.2判断col[j]，如果为false，则将matrix[0~i-1][j]置为0，并且col[j] = true

java代码：

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean[] line = new boolean[m], col = new boolean[n];
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(0 != matrix[i][j]) {
                    matrix[i][j] = (col[j] || line[i]) ? 0 : matrix[i][j];
                    continue;
                }

                if(!line[i]) {
                    for(int k = 0; k < j; k++) {
                        matrix[i][k] = 0;
                    }
                    line[i] = true;
                }
                if(!col[j]) {
                    for(int k = 0; k < i; k++) { 
                        matrix[k][j] = 0;
                    }
                    col[j] = true;
                }
            }
        }
    }
}

2.常数的额外空间

解题思路：

上面的做法是用boolean line[m]和boolean col[n]来记录某一行和某一列是否有0，常数级别的额外空间，则是找到一个0元素所处的行和列，用该行和该列来存储某一行和某一列是否有0（因为该行和该列最后都会全部置为0，所以不会影响最终结果）

java代码：

class Solution {
    public void setZeroes(int[][] matrix) {
        int line = -1, col = -1;
        int m = matrix.length, n = matrix[0].length;
        //找到0元素所处的行(line)和列(col)
        for(int i = 0; i < m; i++) {
            int j = 0;
            while(j < n && 0 != matrix[i][j]) j++;
            if(j < n) {
                line = i;
                col = j;
                break;
            }
        }

        if(-1 == line)
            return;

        //遍历数组，记录哪些行和哪些列含有0元素
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(0 == matrix[i][j]) {
                    matrix[line][j] = 0;
                    matrix[i][col] = 0;
                }
            }
        }

        //遍历数组，将含有0元素所处的行和列置为0，并且将用于临时存储信息的行(line)和列(col)置0
        for(int i = 0; i < m; i++) {
            if(0 == matrix[i][col] && i != line) {
                for(int j = 0; j < n; j++) {
                    matrix[i][j] = 0;
                }
            }
            matrix[i][col] = 0;
        }
        for(int i = 0; i < n; i++) {
            if(0 == matrix[line][i] && i != col) {
                for(int j = 0; j < m; j++) {
                    matrix[j][i] = 0;
                }
            }
            matrix[line][i] = 0;
        }
    }
}