BFS解决FloodFill算法：（Leetcode：733. 图像渲染）

南林yan2024-04-24 13:31

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

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

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

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;
    }
};