LeetCode73.矩阵置零

LeetCode73.矩阵置零_第1张图片

 这道题我感觉还是挺简单的,一下子就想到了,不过我的算法很简单很垃圾,效率很低,我一看完题的想法就是直接遍历一遍数组,然后把为0的元素的行和列都存起来,然后把这些行和列都置零就好了,但是这里会有重复,所以用HashSet来存就好了,然后就写出来了,以下是我的代码:

class Solution {
    public void setZeroes(int[][] matrix) {
        int r = matrix.length;
        int c = matrix[0].length;
       Set row = new HashSet();
       Set column = new HashSet();
       for(int i =0;i

看看题解,题解的第一种方法和我的差不多,但是他的感觉更容易理解,它没有拿两个set去记录要置零的行和列,而是拿两个boolean数组来标记(row标记行,col标记列),先第一遍遍历数组,对于遍历到的元素X,如果它等于0,就把两个标记数组中它对应的行和列改为true;然后第二遍遍历数组,对于遍历到的元素X,如果他对应的行或者列在标记数组中是true,就把这个元素改为0,这样其实只要两遍完整的遍历数组就可以,以下是题解第一种解法的代码:

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

题解的第二种解法就是不另外开辟两个数组来标记,而是拿数组的第一行和第一列来标记,但是这样的话第一行和第一列的数就改变了,所以他创建了两个boolean变量flagCol0和flagRow0来标记第一行中是否有0,先遍历第一行,如果存在一个0就把flagRow0改为true,同理然后遍历第一列,然后去遍历数组中剩下的其他元素,如果是0,就把第一行和第一列中对应坐标下的元素改为0,然后再重新遍历一遍除第一行第一列外的元素,如果这个元素的行和列在标记数组(第一行第一列)中是0,那就把这个元素改成0(和题解1一样,只是标记数组变了),最后再看第一行和第一列的那两个标记变量是不是true,如果是再把第一行和第一列改成0,就可以了。我看完还有一个小疑惑,在第一遍遍历中把第一行和第一列的元素改成了0,如果两个标记变量都是false你没把第一行而第一列中改了的元素改回去啊,但一下子就想通了,如果把第一行或第一列中的元素改成了0说明这个元素在有0的行或列,它本来就要变成0,所以是我想多了,以下是题解二代码:

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean flagCol0 = false, flagRow0 = false;
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                flagCol0 = true;
            }
        }
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) {
                flagRow0 = true;
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }
        if (flagCol0) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
        if (flagRow0) {
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }
    }
}

你可能感兴趣的:(leetcode,矩阵,算法,leetcode,java)