矩阵置零之脑筋急转弯题型的练习

前言

脑筋急转弯型题比其它需要数据结构的题更脱离代码，也就是更抽象，更不依赖于具体的语言，及语言的特有的结构。完全是逻辑上的练习，与代码无关。
笔试：需要多练习这种题，没做过，笔试时就基本上想不到，做不来。
练习目的：这种题的上下限很高很低，磨练思维，提升智力。
注意事项：当然基础的数据结构和经典算法思想是基础，必须掌握。

一、矩阵置零

二、一题多解，打开视野

package com.xhu.offer.top100;

import java.util.ArrayList;
import java.util.List;

//矩阵置零
public class SetZeroes {
    /*
    target:将矩阵中的零元素所在的行和列中的元素全部设为0;
    1-如何知道矩阵中的零元素？通过遍历，用list记录矩阵中0元素的行列坐标。
    2-如何设置0元素所在行列的所有元素为0？通过for循环将行和列的元素赋值为0即可。
    P1：对一行一列赋值时，可能会重复赋值，这样浪费时间。
    S1：通过行列hash来记录已经被覆盖掉的列来减少代码运行时间。
     */
    public void setZeroes(int[][] matrix) {
        //记录0元素所在位置
        List<int[]> zero = new ArrayList<>();
        int m = matrix.length, n = matrix[0].length;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (0 == matrix[i][j]) zero.add(new int[]{i, j});
            }
        }
        //遍历zero List，进行行列覆盖。
        //行列hash
        int[] row = new int[m];
        int[] col = new int[n];

        int size = zero.size();
        for (int i = 0; i < size; i++) {
            int[] cur = zero.get(i);
            //行赋值
            if (0 == row[cur[0]]) {
                for (int j = 0; j < n; j++) {
                    matrix[cur[0]][j] = 0;
                }
                row[cur[0]] = 1;
            }
            //列赋值
            if (0 == col[cur[1]]) {
                for (int j = 0; j < m; j++) {
                    matrix[j][cur[1]] = 0;
                }
                col[cur[1]] = 1;
            }
        }
    }
}

//题目要求：用O(1)的空间复杂度来解决。
class SetZeroes2 {
    /*
    target:以O(1)的空间复杂度来进行0值的行列覆盖。
    把该0元素所在行所在列全部赋值为0，覆盖行，都是从第一列开始覆盖到最后一列；覆盖列，都是从第一行覆盖到最后一行。
    所以，找到0元素，将该行与第一列相交位置标记为0，表示这一行是要被覆盖的；将所在列与第一行相交位置的值标记为，0表示这一列是要被覆盖的。
    P1：那么第一行中有元素为0，则将(0,0)处标记为0，那么这是表示需要覆盖第一行呐？还是覆盖第一列呐？
    S1：用行列标志来记录是行还是列--针对第一个元素
     */
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean row = false, col = false;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (0 == matrix[i][j]) {
                    //是否为第一行
                    if (0 == i) row = true;
                    //是否为第一列
                    if (0 == j) col = true;
                    //做好该行该列的标记为
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        //根据标记来完成行列的覆盖。
        //bug1:index应该从1开始，毕竟第一行是标记位，不能被覆盖，应该留到row|col的情况来做最后的覆盖。
        for (int i = 1; i < m; i++) if (0 == matrix[i][0]) for (int j = 0; j < n; j++) matrix[i][j] = 0;
        for (int i = 1; i < n; i++) if (0 == matrix[0][i]) for (int j = 0; j < m; j++) matrix[j][i] = 0;
        //(0,0)位置元素为0时，看第一行和第一列是否需要覆盖为0
        if (row) for (int i = 0; i < n; i++) matrix[0][i] = 0;
        if (col) for (int i = 0; i < m; i++) matrix[i][0] = 0;
    }
    /*
    总结:先用第一行做位标记位，来覆盖其它地方；再根据(0,0)位置的情况，再覆盖标志位。
        分为前后 需同步 的两步走。
     */
}

总结

笔试：需要多练习这种题，没做过，笔试时就基本上想不到，做不来。
练习目的：这种题的上下限很高很低，磨练思维，提升智力。
注意事项：当然基础的数据结构和经典算法思想是基础，必须掌握。

参考文献

[1] LeetCode 矩阵置零