BFS解决FloodFill算法:(Leetcode:733. 图像渲染)

题目链接:733. 图像渲染 - 力扣(LeetCode)

使用广度优先遍历算法解决该问题:

从初始位置开始搜索,初始位置符合条件就入栈,并修改初始位置值。初始位置出栈。

再从初始位置开始广度优先搜索(上下左右)其他符合条件的位置入栈,并修改它们的值(搜索到即修改,防止同一位置被多次入栈)。每次出栈一个位置,并再次从该位置广度优先搜索,循环直至栈空。

cpp 复制代码
typedef pair<int,int> PII;
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
        //处理边界条件(初始位置即为color则不需要修改)
        if(image[sr][sc]==color)
            return image;
        
        //广度优先搜索
        int prev=image[sr][sc];//记录初始位置的值
        queue<PII> q;
        q.push({sr,sc});//将初始位置入栈
        image[sr][sc]=color;//修改值
        int m=image.size();//行数,纵坐标边界值
        int n=image[0].size();//列数,横坐标边界值
        while(q.size())
        {
            int a=q.front().first;
            int b=q.front().second;
            q.pop();//出栈
            //广度优先搜索
            for(int i=0;i<4;++i)
            {
                int x=a+dx[i];
                int y=b+dy[i];
                if(x>=0&&y>=0&&x<m&&y<n&&image[x][y]==prev)
                {
                    q.push({x,y});
                    image[x][y]=color;//修改值
                }
            }
        }
        return image;
    }
};
相关推荐
码爸4 小时前
线性回归算法
算法·机器学习·线性回归
Coder-magician5 小时前
《代码随想录》刷题打卡day51:图论-part09(dijistra堆优化,Bellman_ford算法)
算法·图论
Nil2085 小时前
leetcode 763划分字母区间
算法·leetcode·职场和发展
不会就选b5 小时前
算法日常・每日刷题--<动态规划>20
算法·leetcode·动态规划
青少儿编程课堂6 小时前
迪杰斯特拉(Dijkstra)算法:单源最短路与赛场路线规划
c++·python·算法·bfs·信息学竞赛
不会就选b6 小时前
算法日常・每日刷题--<动态规划>19
算法·leetcode·职场和发展
没有钱的钱仔7 小时前
协方差矩阵含义
算法·机器学习·矩阵
垆边人似月.7 小时前
华为机试题 :和为 K 的子数组个数
算法·leetcode·华为
三*一8 小时前
优化修整算法
前端·javascript·算法