深度优先搜索LeetCode 733 Flood Fill

LeetCode 733

Flood Fill

• Problem Description：
  题目给出一个矩阵，指定其中一个位置为起始位置，从起始位置开始向四周（上、下、左、右）如洪水般扩散，但扩散区域限于具有与起始位置相同值的点。这些被洪水影响的矩阵中的点会被赋予一个新的值newColor，题目要求返回经过扩散影响后的矩阵。
  具体的题目信息：
  https://leetcode.com/problems/flood-fill/description/
• Example:
  
• Solution:

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int temp = image[sr][sc];
        if (temp == newColor) return image; //此条件必须加上，消除死循环
        Flood(image, sr, sc, newColor, temp);
        return image;
    }
    void Flood(vector<vector<int>>& image, int sr, int sc, int newColor, int temp) {
        if (sr >= 0 && sr < image.size() && sc >= 0 && sc < image[0].size() && image[sr][sc] == temp) {
            image[sr][sc] = newColor;
            Flood(image, sr-1, sc, newColor, temp);
            Flood(image, sr+1, sc, newColor, temp);
            Flood(image, sr, sc-1, newColor, temp);
            Flood(image, sr, sc+1, newColor, temp);
        }
        return;
    }
};