Set Matrix Zeroes Leetcode Solution

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        const int m = matrix.size(); // Get the number of rows
        const int n = matrix[0].size(); // Get the number of columns

        // Flags to check if the first row or column needs to be filled with zeros
        bool shouldFillFirstRow = false;
        bool shouldFillFirstCol = false;

        // Check if the first row has any zero element
        for (int j = 0; j < n; ++j)
            if (matrix[0][j] == 0) {
                shouldFillFirstRow = true;
                break;
            }

        // Check if the first column has any zero element
        for (int i = 0; i < m; ++i)
            if (matrix[i][0] == 0) {
                shouldFillFirstCol = true;
                break;
            }

        // Store the information in the first row and the first column.
        for (int i = 1; i < m; ++i)
            for (int j = 1; j < n; ++j)
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0; // Mark the corresponding element in the first column as 0
                    matrix[0][j] = 0; // Mark the corresponding element in the first row as 0
                }

        // Fill 0s for the matrix except the first row and the first column.
        for (int i = 1; i < m; ++i)
            for (int j = 1; j < n; ++j)
                if (matrix[i][0] == 0 || matrix[0][j] == 0)
                    matrix[i][j] = 0;

        // Fill 0s for the first row if needed.
        if (shouldFillFirstRow)
            for (int j = 0; j < n; ++j)
                matrix[0][j] = 0;

        // Fill 0s for the first column if needed.
        if (shouldFillFirstCol)
            for (int i = 0; i < m; ++i)
                matrix[i][0] = 0;
    }
};