【Lintcode】1062. Flood Fill

题目地址：

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(mn)$。