题目简介：
有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。
给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。
为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
示例 1:
1
2
3
4
5
6
7
8
9
| 输入:
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] 内。- 给出的初始点将满足
0 <= sr < image.length 和 0 <= sc < image[0].length。
image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。
思路：
深度优先搜索，保存原先的颜色以后，在原图的基础上修改即可，
tips:
- 当原图颜色与新颜色相同时，直接返回原图，否则会陷入死循环中。
- 利用
dx, dy数组方便方向上的判断。
代码如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
| class Solution {
public:
int dx[4] = {0, 1, 0, -1};
int dy[4] = {-1, 0, 1, 0};
int oldcolor;
void dfs(vector<vector<int>>& image, int sr, int sc, int newColor){
image[sr][sc] = newColor;
for(int i = 0; i < 4; i++){
int x = sr + dx[i];
int y = sc + dy[i];
if(x >= 0 && x < image.size() && y >= 0 && y < image[0].size()){
if(image[x][y] == oldcolor)
dfs(image, x, y, newColor);
}
}
}
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
if(image[sr][sc] == newColor)
return image;
oldcolor = image[sr][sc];
dfs(image, sr, sc, newColor);
return image;
}
};
|
