没事刷刷LetCode——733.图像渲染

CSDN话题挑战赛第2期
参赛话题:算法题解

1.题目描述

有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

为了完成 上色工作 ,从初始像素开始,记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为 newColor 。

最后返回 经过上色渲染后的图像 。

2.题目链接

733.图像渲染

3.思路讲解

首先我们需要掌握一个方法,使用两个常数数组表达x与y的变化,代表上下左右四个坐标。
x = {0, -1, 0, 1};
y = {1, 0, -1, 0};
另外就是使用BFS(广度优先搜索)

4.模板代码

    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
    
    
    	// 维护上下左右四个领域
        const int dx[4] = {
    
     0,1,0,-1 };
        const int dy[4] = {
    
     1,0,-1,0 };
        // 获得当前颜色
        int currColor = image[sr][sc];
        // 如果当前颜色与目标颜色已经一样了,直接返回
        if (currColor == color) {
    
    
            return image;
        }
        // 获取图片大小
        int n = image.size(), m = image[0].size();
        queue<pair<int, int>> que;
        // 当前位置入队
        que.emplace(sr, sc);
        // 当前位置颜色改变
        image[sr][sc] = color;
        // 当队列不为空,一直广度搜索
        while (!que.empty())
        {
    
    	
        	// 获取上一次改变的像素点的位置坐标
            int x = que.front().first, y = que.front().second;
            // 处理就弹出
            que.pop();
            // 上下左右四个坐标不超界,与当前颜色相等,插入队列
            for (int i = 0; i < 4; i++) {
    
    
                int mx = x + dx[i], my = y + dy[i];
                if (mx >= 0 && mx < n && my >= 0 && my < m && image[mx][my] == currColor) {
    
    
                    que.emplace(mx, my);
                    image[mx][my] = color;
                }
            }
        }
        return image;
    }

猜你喜欢

转载自blog.csdn.net/REstrat/article/details/126910824