颜色填充..-CSDN博客

本文链接：https://blog.csdn.net/zndxj/article/details/126796001

面试题 08.10. 颜色填充
该题的题意为写一个函数来实现许多图片编辑软件都支持的「颜色填充」功能。即为将颜色相同且在上、下、左、右四个方向上存在相连情况的若干元素都染成相同的新颜色。
在解该题时可以想到dfs，即先找到初始坐标点，再进行深搜染色，如果超界或是与原来的颜色不一样则换方向继续。
该题主要涉及dfs，用递归实现。

class Solution {
public:
    const int dx[4] = {1, 0, 0, -1};
    const int dy[4] = {0, 1, -1, 0};
    void dfs(vector<vector<int>>& image, int x, int y, int color, int newColor) {
        if (image[x][y] == color) {
            image[x][y] = newColor;
            for (int i = 0; i < 4; i++) {
                int mx = x + dx[i], my = y + dy[i];
                if (mx >= 0 && mx < image.size() && my >= 0 && my < image[0].size()) {
                    dfs(image, mx, my, color, newColor);
                }
            }
        }
    }

    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int currColor = image[sr][sc];
        if (currColor != newColor) {
            dfs(image, sr, sc, currColor, newColor);
        }
        return image;
    }
};