题目链接: 图像渲染
用 FloodFill 算法求解, FloodFill 算法即为在一个矩阵中找到属性相同的连通块.
题解大致步骤:
1. 先判断 image[sr][sc] 的颜色是否和新颜色 color 一致, 如果一致则直接返回矩阵 image, 因为题目要求将与初始坐标相同的相连像素点变为 color, 那这都一致了就不用找了不用变了, 直接返回就完事了.
2. 定义方向数组, 用于遍历某个坐标的上下左右是不是与其相同颜色.
3. 定义一个队列用于存储坐标 (x, y), 首先将初始坐标加入到队列.
4. 做一次 bfs, 依次将与初始值相连且颜色相同的坐标添加到队列并每次从队列取出数据进行变色.
题解代码:
class Solution
{
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color)
{
int old_color = image[sr][sc];
if(old_color == color) return image;
queue<pair<int, int>> indexs;
indexs.push({sr, sc});
int dx[4] = {0, 0, -1, 1};
int dy[4] = {1, -1, 0, 0};
while(indexs.size())
{
auto [a, b] = indexs.front();
indexs.pop();
image[a][b] = color;
for(int i = 0; i < 4; ++i)
{
int x = a + dx[i];
int y = b + dy[i];
if(x >= 0 && x < image.size() && y >= 0 && y < image[0].size() && image[x][y] == old_color)
{
indexs.push({x, y});
}
}
}
return image;
}
};