Easy 733. Flood Fill

stateDiagram-v2
    [*] --> Start

    state FloodFill {
        Start --> CheckPixel
        CheckPixel --> Fill: 顏色相同
        CheckPixel --> Skip: 顏色不同
        Fill --> Explore: 填充當前像素

        state Explore {
            [*] --> CheckUp
            [*] --> CheckRight
            [*] --> CheckDown
            [*] --> CheckLeft

            CheckUp --> CheckPixel: 有效座標
            CheckRight --> CheckPixel: 有效座標
            CheckDown --> CheckPixel: 有效座標
            CheckLeft --> CheckPixel: 有效座標
        }
    }

    Skip --> [*]
    Explore --> [*]: 所有方向都檢查完

type Image = number[][];

/**
 * 實作 Flood Fill 演算法
 * @param image - 二維陣列代表圖片
 * @param sr - 起始行座標
 * @param sc - 起始列座標
 * @param color - 新的顏色值
 * @returns 填充後的圖片
 */
function floodFill(image: Image, sr: number, sc: number, color: number): Image {
  // 取得圖片尺寸
  const rows = image.length;
  const cols = image[0].length;

  // 取得原始顏色
  const originalColor = image[sr][sc];

  // 如果新舊顏色相同，直接返回
  if (originalColor === color)
    return image;

  // DFS 填充函數
  const fill = (r: number, c: number): void => {
    // 檢查邊界和顏色
    if (
      r < 0 || r >= rows
      || c < 0 || c >= cols
      || image[r][c] !== originalColor
    ) {
      return;
    }

    // 填充當前像素
    image[r][c] = color;

    // 遞迴填充四個方向
    fill(r - 1, c); // 上
    fill(r + 1, c); // 下
    fill(r, c - 1); // 左
    fill(r, c + 1); // 右
  };

  // 開始填充
  fill(sr, sc);
  return image;
}