题解 | #牛群避免水洼#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param matrix int整型二维数组
     * @return int整型二维数组
     */
    public int[][] avoidPuddles (int[][] matrix) {
        //  matrix[i][j] = 0 表示在 i 行和 j 列有一个水洼。
        // 对于每个水洼所在的行和列的所有元素，将矩阵元素设为 -1。
        int m = matrix.length;
        int n = matrix[0].length;
        boolean [] flagX = new boolean[m];
        boolean [] flagY = new boolean[n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    // 是水洼
                    flagX[i] = true;
                    flagY[j] = true;
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (flagX[i] == true || flagY[j] == true) {
                    matrix[i][j] = -1;
                }
            }
        }
        return matrix;
    }
}