【程序员面试金典】 08.10. 颜色填充(DFS)

1.题目

颜色填充。编写函数，实现许多图片编辑软件都支持的“颜色填充”功能。给定一个屏幕（以二维数组表示，元素为颜色值）、一个点和一个新的颜色值，将新颜色值填入这个点的周围区域，直到原来的颜色值全都改变。

示例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]内。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/color-fill-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

这个题有个比较tricky的地方，就是当目标(sr, sc)坐标位置的颜色与newColor一样时，就不能再进行dfs遍历。不知道为啥有这个限制。

帮助大家理解题意，该题的意思就是说，将选中的元素【sr ， sc】与新的颜色进行比较

• 如果想同，则不作任何操作。
• 如果不同，记录原有元素的颜色，然后向上下左右四个方向进行搜索，更改颜色为就颜色的元素为新颜色。

class Solution
{
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor)
    {
        row = image.size(), col = image[0].size();
        int oldColor = image[sr][sc];
        if(color!=newColor) DFS(image, sr, sc, newColor, oldColor);
        return image;
    }
    void DFS(vector<vector<int>>& image, int sr, int sc, int newColor, int oldColor)
    {
        image[sr][sc] = newColor;
        if(sr+1 <row && image[sr+1][sc] == oldColor) DFS(image, sr+1, sc, newColor, oldColor);
        if(sr-1 >=0 && image[sr-1][sc] == oldColor)  DFS(image, sr-1, sc, newColor, oldColor);
        if(sc+1 <col && image[sr][sc+1] == oldColor) DFS(image, sr, sc+1, newColor, oldColor);
        if(sc-1 >=0 && image[sr][sc-1] == oldColor)  DFS(image, sr, sc-1, newColor, oldColor);
    }
private:
    int row, col;
};