题解 | #牛群避免水洼# 模拟

知识点

模拟

思路

遍历每一个位置，如果这个位置是0，那么给该行和该列打个标记。最后给每个有标记的行和列标记为-1

时间复杂度

AC Code（C++）

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param matrix int整型vector<vector<>> 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > avoidPuddles(vector<vector<int> >& matrix) {
        int n = matrix.size(), m = matrix[0].size();
        vector<bool> row(n, false), col(m, false);
        for (int i = 0; i < n; i ++) {
            for (int j = 0; j < m; j ++) {
                if (matrix[i][j] == 0) {
                    row[i] = true;
                    col[j] = true;
                }
            }
        }
        for (int i = 0; i < n; i ++) {
            for (int j = 0; j < m; j ++) {
                if (row[i] or col[j]) matrix[i][j] = -1;
            }
        }
        return matrix;
    }
};