leetcode733. 图像渲染/dfs,bfs

题目:733. 图像渲染

有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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,
因为它不是在上下左右四个方向上与初始点相连的像素点。

注意:

  • 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/flood-fill
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

基本思想:dfs

class Solution {
    
    
public:
    vector<vector<int>> dir;
    int row, col, pre_color;
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
    
    
        dir = {
    
    {
    
    -1, 0}, {
    
    0, -1}, {
    
    0, 1}, {
    
    1, 0}};
        row = image.size();
        col = image[0].size();
        pre_color = image[sr][sc];
        image[sr][sc] = newColor;
        if(newColor != pre_color)//这样处理的目的是为了避免出现当颜色相等的时候出现循环递归
            dfs(image, sr, sc, newColor);
        return image;
    }
    void dfs(vector<vector<int>>& image, int x, int y, int color){
    
    
        for(auto d : dir){
    
    
            int cx = x + d[0];
            int cy = y + d[1];
            if(judge(cx, cy) && image[cx][cy] == pre_color){
    
    
                image[cx][cy] = color;
                dfs(image, cx, cy, color);
            }
        }
    }
    bool judge(int x, int y){
    
    
        return x >= 0 && x < row && y >= 0 && y < col;
    }
};

当然这道题目也可以用bfs来写,将与起始位置相同的点入队,同时更新起始位置的颜色,直到队列为空的时候结束。

当然这道题要注意,起始位置的颜色和要更新的颜色相同的时候,无需处理,如果在此不加以判断,会造成死循环。

猜你喜欢

转载自blog.csdn.net/qq_31672701/article/details/108036151