LeetCode-73. 矩阵置零-Java-medium

题目链接

法一

    /**
     * 时间复杂度 O(m * n)
     * 空间复杂度 O(1)
     *
     * @param matrix
     */
    public void setZeroes(int[][] matrix) {
        int row = matrix.length, col = matrix[0].length;
        if (row == 0 || col == 0) {
            return;
        }
        boolean rowFlag = false, colFlag = false;
        for (int i = 0; i < row; i++) { // 标记第1列是否是有0
            if (matrix[i][0] == 0) {
                colFlag = true;
                break;
            }
        }
        for (int j = 0; j < col; j++) { // 标记第1行是否是有0
            if (matrix[0][j] == 0) {
                rowFlag = true;
                break;
            }
        }
        for (int i = 1; i < row; i++) { // 遍历除第1行第1列以外的矩阵元素
            for (int j = 1; j < col; j++) {
                if (matrix[i][j] == 0) { // 如果有元素为0，则将对应的第1行第1列的位置置为0
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < row; i++) {  // 遍历除第1行第1列以外的矩阵元素
            for (int j = 1; j < col; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) { // 如果matrix[i][j]所在位置对应的第1行第1列有任意一个位置为0
                    matrix[i][j] = 0; // matrix[i][j]置为0
                }
            }
        }
        if (colFlag) { // 第1列是否有0 ，有则将该列置为0
            for (int i = 0; i < row; i++) {
                matrix[i][0] = 0;
            }
        }
        if (rowFlag) { // 第1行是否有0 ，有则将该行置为0
            for (int j = 0; j < col; j++) {
                matrix[0][j] = 0;
            }
        }
    }

本地测试

        /**
         * 73. 矩阵置零
         */
        lay.showTitle(73);
        Solution73 sol73 = new Solution73();
        int[][] matrix73 = new int[][]{{0, 1, 2, 0}, {3, 4, 5, 2}, {1, 3, 1, 5}};
        arrayOpt.showIntTwoDimArray(matrix73, matrix73.length);
        sol73.setZeroes(matrix73);
        arrayOpt.showIntTwoDimArray(matrix73, matrix73.length);