图像渲染(DFS)

24 阅读1分钟

 经典老模板题了,直接上代码

对了这里有个小坑,就是可能出现image[sr][sc]已经是newcolor,所以我们需要先判断一下再搜索

class Solution {
public:
    int nextt[4][2]={{0,1},{1,0},{0,-1},{-1,0}},tx,ty;
    vector<vector<int>> floodFill(vector<vector<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;
    }
    void dfs(vector<vector<int>>& image, int x, int y, int newColor,int oldcolor)
    {   
        if(image[x][y]==oldcolor)
        {
            image[x][y]=newColor;
            for(int i=0;i<4;i++)
            {
                tx=x+nextt[i][0];
                ty=y+nextt[i][1];
                if(tx<0||ty<0||tx>=image.size()||ty>=image[0].size())
                continue;
                dfs(image,tx,ty,newColor,oldcolor);
            }
        }
        return ;
    }

};

下面这个是题解(和我写法不同而已) 

class Solution {
public:
    int n, m, w;
    void DFS(vector<vector<int>>& image, int x, int y, int c){
        if(x<0 || x>=n || y<0 || y>=m || image[x][y]!=w || image[x][y]==c) // 退出条件:位置越界 或 当前像素与起始点不同 或 当前像素与新颜色相同
            return;
        image[x][y] = c; // 渲染当前位置像素为新颜色值
        DFS(image, x+1, y, c); // 深搜4个方向
        DFS(image, x-1, y, c);
        DFS(image, x, y+1, c);
        DFS(image, x, y-1, c);
    }
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        n=image.size();
        m=image[0].size();
        w=image[sr][sc];
        if(w != newColor) // 起点像素与新颜色值不同则以此为起点开始深搜
            DFS(image, sr, sc, newColor);
        return image;
    }
};