설명

예시 1

입력

matrix = [
    [0, 1, 0],
    [1, 1, 1],
    [1, 1, 1]
]

산출

[
    [0, 1, 0],
    [1, 2, 1],
    [2, 3, 2]
]

직관

구현

int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};

vector<vector<int>> solve(vector<vector<int>>& matrix) {
    int rows = matrix.size(), cols = matrix[0].size();
    vector<vector<bool>> visited(rows, vector<bool>(cols, false));
    queue<pair<int, int>> q;
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cols; j++) {
            if (matrix[i][j] == 0) q.push({i, j});
        }
    }
    int val = 0;
    while (!q.empty()) {
        int size = q.size();
        for (int i = 0; i < size; i++) {
            auto point = q.front();
            q.pop();
            if (visited[point.first][point.second]) continue;
            visited[point.first][point.second] = true;
            matrix[point.first][point.second] = val;
            for (int j = 0; j < 4; j++) {
                int x = point.first + dx[j], y = point.second + dy[j];
                if (x >= 0 && x < rows && y >= 0 && y < cols && !visited[x][y]) q.push({x, y});
            }
        }
        val++;
    }
    return matrix;
}

시간 복잡도