算法题十三 之 图像渲染

题目

leetcode 传送门

我发现，原题描述很让人迷惑，把一道简简单单的题目故意说得很复杂。

简单翻译一下

• 有一个矩阵，宽高是 n 与 m
• 这个矩阵里有颜色值，有一个数字表示
• 现在给你一个点（sr，sc）和一个新的颜色值，让你把这个矩阵与点（sr，sc）直接与间接相连格子染成新的颜色值。
• 判断相连的方式是 上 下 左 右

思路

1. 从起点（sr，sc）开始深度遍历
2. 每次对比上下左右的颜色值是否与当前点一致
3. 如果一致，则递归调用
4. 如果不一致，则停止。

附上代码

public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
    if(image[sr][sc] == newColor){
        return image;
    }
    dfs(image,sr,sc,newColor);
    return image;
}

private void dfs(int[][] image,int i,int j,int newColor){
    int old = image[i][j];
    image[i][j] = newColor;
    if(i > 0 && image[i - 1][j] == old){
        dfs(image,i - 1,j,newColor);
    }
    if(i < image.length - 1 && image[i + 1][j] == old){
        dfs(image,i + 1,j,newColor);
    }
    if(j > 0 && image[i][j - 1] == old){
        dfs(image,i,j - 1,newColor);
    }
    if(j < image[i].length - 1 && image[i][j + 1] == old){
        dfs(image,i,j + 1,newColor);
    }
}