题解 | #牛群避免水洼#

知识点：矩阵

题目要求若一行或一列内有一个位置的元素为0，则整行或整列的元素都应该修改为-1，要求使用原地算法，故我们可以先使用额外的数组来统计那些行与列的元素为0，具体来说，使用一个行数组和一个列数组，分别记录该行或该列是否有0元素，遍历完成后，进行二次遍历，如当前行或列已经被标记，则需要将数组的元素置为-1。

Java题解如下：

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param matrix int整型二维数组 
     * @return int整型二维数组
     */
    public int[][] avoidPuddles (int[][] matrix) {
        // write code here
        int m = matrix.length;
        int n = matrix[0].length;
        boolean[] row = new boolean[m];
        boolean[] column = new boolean[n];
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(matrix[i][j] == 0) {
                    row[i] = true;
                    column[j] = true;
                }
            }
        }
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(row[i] || column[j]) {
                    matrix[i][j] = -1;
                }
            }
        }
        return matrix;
    }
}