题目地址:
https://www.lintcode.com/problem/flood-fill/description
给定一个二维矩阵,每个位置的数字代表一种颜色。给定一个初始位置,再给定一个新的颜色,要求将与其连通的相同颜色的点都设置为那个新的颜色。连通路径允许每一步走上下左右的一格。
可以用DFS。从初始点出发沿着四个方向做DFS,将相同颜色的点都染色为新的颜色即可。代码如下:
public class Solution {
/**
* @param image: a 2-D array
* @param sr: an integer
* @param sc: an integer
* @param newColor: an integer
* @return: the modified image
*/
public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
// Write your code here
if (image == null || image.length == 0 || image[0].length == 0) {
return image;
}
dfs(image, sr, sc, image[sr][sc], newColor);
return image;
}
private void dfs(int[][] image, int x, int y, int oldColor, int newColor) {
// 染色
image[x][y] = newColor;
int[] d = {-1, 0, 1, 0, -1};
// 开始遍历四个方向
for (int i = 0; i < d.length - 1; i++) {
int nextX = x + d[i], nextY = y + d[i + 1];
if (inBound(image, nextX, nextY) && image[nextX][nextY] == oldColor) {
dfs(image, nextX, nextY, oldColor, newColor);
}
}
}
private boolean inBound(int[][] image, int x, int y) {
return 0 <= x && x < image.length && 0 <= y && y < image[0].length;
}
}
时空复杂度 O ( m n ) O(mn) O(mn)。