No.194 - LeetCode733. Flood Fill - BFS或DFS模版题 - 核心思想染色

ShellDawn

于 2020-07-17 17:35:42 发布

阅读量158

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode dfs bfs

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/107413855

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种基于广度优先搜索的二维图像洪水填充算法实现。通过使用队列和标志矩阵来跟踪已处理的像素，确保了算法的正确性和效率。文章详细展示了算法的C++代码实现，包括如何遍历图像、更新像素颜色以及边界检查。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        const int dirx[] = {-1,0,1,0};
        const int diry[] = {0,-1,0,1};
        int a = image.size();
        int b = image[0].size();
        vector<vector<int>> flag(a,vector<int>(b,0));
        queue<pair<int,int>> q;
        q.push(make_pair(sr,sc));
        flag[sr][sc] = 1;
        while(!q.empty()){
            int x = q.front().first;
            int y = q.front().second;
            flag[x][y] = 1;
            for(int i=0;i<4;i++){
                int xx = x+dirx[i];
                int yy = y+diry[i];
                if(xx < 0 || yy < 0 || xx >= a || yy >= b){
                    continue;
                }
                if(flag[xx][yy] == 0 && image[x][y] == image[xx][yy]){
                    q.push(make_pair(xx,yy));
                }
            }
            image[x][y] = newColor;
            q.pop();
        }
        return image;
    }
};