题解 | #牛群避免水洼#

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