面试题 08.10. 颜色填充-----深度优先搜索
题目表述
编写函数,实现许多图片编辑软件都支持的「颜色填充」功能。
待填充的图像用二维数组 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 ,因为它不属于初始坐标点的周围区域。
深度优先搜索
根据题目描述,只需要找起始节点四周相同颜色的块,然后将块染成新的颜色即可。递归的找可以被染色的块,用visited数组记录哪些块已经被访问。同理,普通的广度优先搜索也可以解该题。
class Solution {
int[][] dirs = {{0,1}, {0,-1}, {1,0}, {-1,0}};
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
boolean[][] visited = new boolean[image.length][image[0].length];
dfs(image, sr,sc,newColor,visited);
return image;
}
public void dfs(int[][] image, int sr, int sc, int newColor,boolean[][] visited){
if(sr < 0 || sc< 0 || sr >=image.length || sc >= image[0].length){
return ;
}
if(visited[sr][sc]) return;
visited[sr][sc] = true;
for(int[] dir : dirs){
int x = sr + dir[0];
int y = sc + dir[1];
if(x < 0 || y< 0 || x >=image.length || y >= image[0].length){
continue;
}
if(image[x][y] != image[sr][sc]) continue;
dfs(image, x, y, newColor,visited);
}
image[sr][sc] = newColor;
}
}