程序员面试金典-面试题 08.10. 颜色填充
题目:
颜色填充。编写函数,实现许多图片编辑软件都支持的“颜色填充”功能。给定一个屏幕(以二维数组表示,元素为颜色值)、一个点和一个新的颜色值,将新颜色值填入这个点的周围区域,直到原来的颜色值全都改变。
示例1:
输入:
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]内。
分析:
dfs搜索,不过注意的是,这个填充是在原来的数字相连所填充的,所以除边界条件外,还需要判断新填充位置的颜色是不是原来的颜色,如果不是就返回。
程序:
class Solution { public int[][] floodFill(int[][] image, int sr, int sc, int newColor) { m = image.length; n = image[0].length; int[][] visit = new int [m][n]; int oldColor = image[sr][sc]; dfs(image, visit, sr, sc, newColor, oldColor); return image; } private void dfs(int[][] image, int[][] visit, int x, int y, int newColor, int oldColor){ if(x < 0 || x >= m || y < 0 || y >= n || visit[x][y] == 1 || image[x][y] != oldColor) return; image[x][y] = newColor; visit[x][y] = 1; for(int i = 0; i < 4; ++i){ int nx = x + move[i][0]; int ny = y + move[i][1]; dfs(image, visit, nx, ny, newColor, oldColor); } } private int m; private int n; private int[][] move = new int[][]{{1, 0}, {0, 1}, {-1, 0}, {0, -1}}; }