733. 图像渲染

88 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情

一、题目描述:

733. 图像渲染 - 力扣(LeetCode)

有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 srscnewColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

为了完成 上色工作 ,从初始像素开始,记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为 newColor 。

最后返回 经过上色渲染后的图像 。

示例 1:

image.png

输入: image = [[1,1,1],[1,1,0],[1,0,1]],sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析: 在图像的正中间,(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
注意,右下角的像素没有更改为2,因为它不是在上下左右四个方向上与初始点相连的像素点。

示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]

 

提示:

  • m == image.length
  • n == image[i].length
  • 1 <= m, n <= 50
  • 0 <= image[i][j], newColor < 216
  • 0 <= sr < m
  • 0 <= sc < n

二、思路分析:

准备一个集合存储所有相连的像素点,集合初始值为所给定的初始坐标

遍历集合每个坐标上下左右四个方向所有相连的像素点,每次有新的像素就加入到集合中,直到集合不在有新的元素加入停止遍历

优化:新增一个新元素集合保存每次遍历的新增的元素,下一次遍历只需要遍历该新元素集合

三、AC 代码:

class Solution {

    /**
     * @param Integer[][] $image
     * @param Integer $sr
     * @param Integer $sc
     * @param Integer $newColor
     * @return Integer[][]
     */
    function floodFill($image, $sr, $sc, $newColor) {
        $width = count($image[0]);
        $height = count($image);
        $oldColor = $image[$sr][$sc];
        $result[] = $sr.','.$sc;
        $new = $result;
        $last = [];
        do {
            $preNum = count($result);
            $last = $result;

            foreach ($new as $val) {
                list($r, $c) = explode(',', $val);

                //上
                for ($i = $r-1; $i >= 0; $i--) {
                    if ($oldColor == $image[$i][$c]) {
                        $coord = $i.','.$c;
                        if (!in_array($coord, $result)) {
                            $result[] = $coord;
                        }
                    } else {
                        break;
                    }
                }

                //下
                for ($i = $r+1; $i < $height ; $i++) {
                    if ($oldColor == $image[$i][$c]) {
                        $coord = $i.','.$c;
                        if (!in_array($coord, $result)) {
                            $result[] = $coord;
                        }
                    } else {
                        break;
                    }
                }

                //左
                for ($i = $c-1; $i >= 0 ; $i--) {
                    if ($oldColor == $image[$r][$i]) {
                        $coord = $r.','.$i;
                        if (!in_array($coord, $result)) {
                            $result[] = $coord;
                        }
                    } else {
                        break;
                    }
                }

                //右
                for ($i = $c+1; $i < $width ; $i++) {
                    if ($oldColor == $image[$r][$i]) {
                        $coord = $r.','.$i;
                        if (!in_array($coord, $result)) {
                            $result[] = $coord;
                        }
                    } else {
                        break;
                    }
                }
            }

            $new = array_diff($result, $last);
        } while ($preNum < count($result));

        foreach ($result as $val) {
            list($r, $c) = explode(',', $val);
            $image[$r][$c] = $newColor;
        }
        return $image;
    }
}