CareerCup 1.7

1.7 Write an algorithm such that if an element in an M*N matrix is 0, its entire row and column are set to 0.

void setZeros(vector<vector<int> >& matrix) {
    int m = matrix.size();
    int n = matrix[0].size();

    vector<bool> row(m, false);
    vector<bool> column(n, false);

    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] = 0;
            }
        }
    }
}