题目: 编写函数,实现许多图片编辑软件都支持的「颜色填充」功能。
待填充的图像用二维数组 image 表示,元素为初始颜色值。初始坐标点的横坐标为 sr 纵坐标为 sc。需要填充的新颜色为 newColor 。周围区域」是指颜色相同且在上、下、左、右四个方向上存在相连情况的若干元素。
请用新颜色填充初始坐标点的周围区域,并返回填充后的图像。
示例:
输入:
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 ,因为它不属于初始坐标点的周围区域。
提示:
- image 和 image[0] 的长度均在范围 [1, 50] 内。
- 初始坐标点 (sr,sc) 满足 0 <= sr < image.length 和 0 <= sc < image[0].length 。
- image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535] 内。
相关题目:
思路: 经典DFS搜索即可:
- 从给定的位置开始搜索,上,下,左,右不断回溯搜索。
- 越过边界或搜索完毕退出。
示例的涂染过程:
DFS思路在两个相关题目中有详细的阐述,这里就不再说了。
1. 自己写的,处理太过冗余:
void dp(vector<vector<int>>& image, int sr, int sc, int newColor,int temp)
{
if(image[sr][sc]==temp)
return;
//上
if((sr-1)>=0 && (sr-1)<image.size() && sc>=0 && sc< image[0].size() && image[sr-1][sc]==temp)
{
image[sr-1][sc]=newColor;
dp(image,sr-1,sc,newColor,temp);
}
//下
if((sr+1