LeetCode——73. 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.

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)的空间复杂度，但是可以优化为O(m + n)，最终还可以优化为常数级空间复杂度，我觉得有点不可思议！
　　我刚开始居然有个很蠢的想法，不知道你们会不会想到。我在想，我直接遍历进去，如果这个元素为０，那我直接将它的行和列都设置为０，不就可以了吗？为什么还要多余的空间呢？后来想了想，觉得这个想法真的是很蠢。
　　后来静下来想想，很容易就想到了O(m + n)空间复杂度的解法：直接将需要置０的行和列的序号保存进一个list中，然后最后再一个一个取出来，在原有的矩阵上将保存的行和列设为０，就搞定了，很简单。一下是代码：
　　

        //用来保存需要置0的行和列
        Set<Integer> rows = new HashSet<>();
        Set<Integer> columns = new HashSet<>();
        //先找到需要置0的行和列
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[0].length; j++) {
                if(matrix[i][j] == 0) {
                    rows.add(i);
                    columns.add(j);
                }
            }
        }
        //然后根据需要置0的行和列来循环二维数组，在相应行和列置0
        for(int i : rows) {
            for(int j = 0; j < matrix[0].length; j++)
                matrix[i][j] = 0;
        }
        for(int j : columns) {
            for(int i = 0; i < matrix.length; i++)
                matrix[i][j] = 0;
        }

　　对吧，其实这个方法还是很简单的。但是使用常数级的空间，还是很有难度的，至少我一开始是没有想到的。后来想了想，既然要标记需要置０的行和列的序号，那我为什么要额外开一个参数来保存呢，我可以直接在矩阵上做标记啊。于是就想到了，可以用第一行和第一列作为一个坐标轴，需要置０的行或者列就在这个坐标轴上标出来。在遍历的同时也判断第一行和第一列是否需要置０，使用一个标记就好了。
　　下面是代码：

        //常数级空间的做法就是用第一行和第一列中的元素来标记需要置0的行和列，这样子就节省了空间
        //最后再根据参数来判断第一行和第一列是否需要置0
        boolean isFirstRowZero = false;
        boolean isFirstColumnZero = false;
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[0].length; j++) {
                if(matrix[i][j] == 0) {
                    //判断首行和首列是否为0
                    if(i == 0)
                        isFirstRowZero = true;
                    if(j == 0)
                        isFirstColumnZero = true;
                    //如果这个元素为0，则在首行和首列进行标记行和列
                    matrix[0][j] = 0;
                    matrix[i][0] = 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;
            }
        }
        //如果首行需要设置为0
        if(isFirstRowZero) {
            for(int j = 0; j < matrix[0].length; j++)
                matrix[0][j] = 0;
        }
        //如果首列需要设置为0
        if(isFirstColumnZero) {
            for(int i = 0; i < matrix.length; i++)
                matrix[i][0] = 0;
        }

　　谢谢大家观看我的博客。如果有不明白的地方，或者是文中有错误的地方，欢迎指出，谢谢！如果大家喜欢我的博客，也可以给我点点赞。你们的点赞就是我的动力！