Flood fill в матрице

EasyАлгоритмы
04:00
Лучше работает на десктопе
AlgorithmsDFSBFSMatrix
Реальный собес1:10-8:50Interview самокат.m4aСтраница собеса

Дана матрица целых чисел matrix, где число обозначает цвет клетки. Также даны координаты стартовой клетки x, y и новый цвет new_color.

Перекрасьте всю 4-связную область стартовой клетки в new_color: можно ходить вверх, вниз, влево и вправо только по клеткам исходного цвета стартовой клетки.

Верните измененную матрицу.

Сигнатура

def flood_fill(matrix: list[list[int]], x: int, y: int, new_color: int) -> list[list[int]]:

Примеры

Пример 1

Вход:
matrix = [[1,1,0],[1,0,0]]
x = 0
y = 0
new_color = 2
Выход:[[2,2,0],[2,0,0]]

Пример 2

Вход:
matrix = [[1,0],[0,1]]
x = 0
y = 0
new_color = 9
Выход:[[9,0],[0,1]]

Диагональ не связана

Пример 3

Вход:
matrix = [[3]]
x = 0
y = 0
new_color = 7
Выход:[[7]]
Консоль
Нажмите Run или Ctrl+Enter для запуска