cci-Q1.7 二维数组置0

原文：

Write an algorithm such that if an element in an MxN matrix is 0, its entire row and column is set to 0.

译文：

写一个函数处理一个MxN的矩阵，如果矩阵中某个元素为0，那么把它所在的行和列都置为0.

不能直接进行遍历置0，因为在遍历的过程中可能遇到之前被置0的元素而不是矩阵元素本身。首先进行遍历，记录下存在0的行和列，然后根据记录的数组重新遍历矩形进行置0操作。

 public static int[][] setZero(int[][] matrix) {
        if (matrix == null) {
            return null;
        }

        int row = matrix.length;
        int column = 0;
        if (row > 0) {
            column = matrix[0].length;
        }

        int[] rows = new int[row];
        int[] columns = new int[column];

        for (int i = 0; i < row; i++) {
            for (int j = 0; j < column; j++) {
                if (0 == matrix[i][j]) {
                    rows[i] = 1;
                    columns[j] = 1;
                }
            }
        }

        for (int i = 0; i < row; i++) {
            for (int j = 0; j < column; j++) {
                if (1 == rows[i] || 1 == columns[j]) {
                    matrix[i][j] = 0;
                }
            }
        }

        return matrix;
    }

junit testcase

  @Test
    public void testSetZero() {
        System.out.println("setZero");
        int[][] test = null;
        int[][] expected = null;
        assertArrayEquals(expected, q1_7.setZero(test));
        int[][] test1 = {};
        int[][] expected1 = {};
        assertArrayEquals(expected1, q1_7.setZero(test1));
        int[][] test2 = {{1, 1, 1}, {1, 0, 1}, {1, 1, 1}};
        int[][] expected2 = {{1, 0, 1}, {0, 0, 0}, {1, 0, 1}};
        assertArrayEquals(expected2, q1_7.setZero(test2));
        int[][] test3 = {{1, 0, 1, 1, 1}, {1, 1, 1, 0, 1}, {1, 1, 1, 1, 1}};
        int[][] expected3 = {{0, 0, 0, 0, 0}, {0, 0, 0, 0, 0}, {1, 0, 1, 0, 1}};
        assertArrayEquals(expected3, q1_7.setZero(test3));
    }