leetcode.cn_面试题 08.10. 颜色填充_2649

leetcode.cn_面试题 08.10. 颜色填充_2649_第1张图片

class Solution {
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        // 新值和旧值相等没有必要进行填充
        if (newColor == image[sr][sc]){
            return image;
        }
        dfs(image,sr,sc,newColor,image[sr][sc]);
        return image;
    }
    private void dfs(int[][] image, int sr, int sc, int newColor,int oldColor){
        // 越界或者这个点不能涂就返回
        if ( sr < 0 || sr >= image.length || sc < 0 || sc >= image[sr].length || image[sr][sc] != oldColor){
            return;
        }

        // 涂色 + 四个方向扩散
        image[sr][sc]  = newColor;
        dfs(image,sr - 1,sc,newColor,oldColor);
        dfs(image,sr + 1,sc,newColor,oldColor);
        dfs(image,sr ,sc - 1 ,newColor,oldColor);
        dfs(image,sr ,sc + 1,newColor,oldColor);
    }
}

你可能感兴趣的:(算法,leetcode,深度优先,算法)