73 Set Matrix Zeroes

题目链接：https://leetcode.com/problems/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(m + n) 空间，并不是最好的解法，但还是记录下来。
使用两个 HashSet 来分别保存存在 0 值的行号和列号。
即，遍历数组将为 0 的元素的行号和列号分别保存在两个 HashSet 中。
在分别取出 HashSet 中保存的行号和列号，将对应行和列的所有元素置 0.

题目中提示，最好的解法是空间复杂度只为常数空间。上网看到了大神的解法，觉得很妙。特别记录下来。

1. 只需要两个变量来记录第一行和第一列是否存在 0 元素。
2. 其它行和列把它们是否需要置 0 （该行或该列存在 0 元素）标记在第一行和第一列中。
3. 然后就是第一行和第一列，如果要置 0，就把它的值赋成0（反正它最终也该是0，无论第一行或者第一列有没有0），否则保留原值。
4. 然后根据第一行和第一列的记录对其他元素进行置 0。
5. 最后再根据前面的两个标记来确定是不是要把第一行和第一列置 0 就可以了。

这样的做法只需要两个额外变量，所以空间复杂度是 O(1)。

大神参考链接：http://blog.csdn.net/linhuanmars/article/details/24066199/

空间复杂度为 O(m + n) 的代码：

public class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix == null || matrix.length == 0)
            return;
        int m = matrix.length;
        int n = matrix[0].length;
        HashSet<Integer> row = new HashSet();
        HashSet<Integer> column = new HashSet();
        for(int i = 0; i < m; i ++) {
            for(int j = 0; j < n; j ++) {
                if(matrix[i][j] == 0) {
                    if(!row.contains(i))
                        row.add(i);
                    if(!column.contains(j))
                        column.add(j);
                }
            }
        }
        for(Integer i : row) {
            for(int j = 0; j < n; j ++)
                matrix[i][j] = 0;
        }
        for(Integer j : column) {
            for(int i = 0; i < m; i ++)
                matrix[i][j] = 0;
        }
        return;
    }
}

157 / 157 test cases passed.
Status: Accepted
Runtime: 5 ms

空间复杂度为 O(1) 的代码：

public class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix == null || matrix.length == 0)
            return;
        boolean rowFlag = false;
        boolean columnFlag = false;
        for(int i = 0; i < matrix.length; i ++) {
            if(matrix[i][0] == 0) {
                columnFlag = true;
                break;
            }
        }
        for(int i = 0; i < matrix[0].length; i ++) {
            if(matrix[0][i] == 0) {
                rowFlag = true;
                break;
            }
        }
        for(int i = 1; i < matrix.length; i ++) {
            for(int j = 1; j < matrix[0].length; j ++) {
                if(matrix[i][j] == 0) {
                    matrix[0][j] = 0;
                    matrix[i][0] = 0;
                }
            }
        }
        for(int i = 1; i < matrix.length; i ++) {
            for(int j = 1; j < matrix[0].length; j ++) {
                if(matrix[0][j] == 0 || matrix[i][0] == 0)
                    matrix[i][j] = 0;
            }
        }
        if(rowFlag) {
            for(int i = 0; i < matrix[0].length; i ++)
                matrix[0][i] = 0;
        }
        if(columnFlag) {
            for(int i = 0; i < matrix.length; i ++)
                matrix[i][0] = 0;
        }
    }
}

157 / 157 test cases passed.
Status: Accepted
Runtime: 2 ms