探索 队列和栈 图像渲染dfs解法

有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
给你一个座标 (sr, sc) 表示图像渲染开始的像素值(行 ,列)和一个新的颜色值 newColor,让你重新上色这幅图像。
为了完成上色工作,从初始座标开始,记录初始座标的上下左右四个方向上像素值与初始座标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始座标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
最后返回经过上色渲染后的图像。
示例 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,
因为它不是在上下左右四个方向上与初始点相连的像素点。
注意:
1.image 和 image[0] 的长度在范围 [1, 50] 内。
2.给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
3.image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

思路:使用dfs解法进行深度优先搜索即可

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
    int color = image[sr][sc];//目标点原有的颜色
    if(newColor == color)//目标点原有的颜色和新的颜色相同
    {}
    else
    dfs(image,sr,sc,newColor,color);    
    return image;
    }
    void dfs(vector<vector<int>>& image, int sr, int sc, int newColor,int color){
        int hang=image.size();
        int lie=image[0].size();
        image[sr][sc]=newColor;
        if(sr-1>=0 && image[sr-1][sc]==color){
            dfs(image,sr-1,sc,newColor,color);
        }
        if(sr+1<hang && image[sr+1][sc]==color){
            dfs(image,sr+1,sc,newColor,color);
        }
        if(sc-1>=0 && image[sr][sc-1]==color){
            dfs(image,sr,sc-1,newColor,color);
        }
        if(sc+1<lie && image[sr][sc+1]==color){
            dfs(image,sr,sc+1,newColor,color);
        }
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章