Solution to #11

bharatr21 · May 11, 2020 · c160cb4 · c160cb4
1 parent b59da30
commit c160cb4
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Solutions/S11.cpp b/Solutions/S11.cpp
@@ -0,0 +1,19 @@
+//Utility to actually flood fill the 2D matrix
+void floodFillUtil(vector<vector<int>>& image, int sr, int sc, int m, int n, int prevColor, int newColor) {
+    if(sr < 0 || sr >= m || sc < 0 || sc >= n) return;
+    else if(image[sr][sc] == newColor || image[sr][sc] != prevColor) return;
+    image[sr][sc] = newColor;
+    floodFillUtil(image, sr - 1, sc, m, n, prevColor, newColor);
+    floodFillUtil(image, sr + 1, sc, m, n, prevColor, newColor);
+    floodFillUtil(image, sr, sc - 1, m, n, prevColor, newColor);
+    floodFillUtil(image, sr, sc + 1, m, n, prevColor, newColor);
+}
+
+//Wrapper function
+vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
+    int prevColor = image[sr][sc];
+    int m = image.size();
+    int n = image[0].size();
+    floodFillUtil(image, sr, sc, m, n, prevColor, newColor);
+    return image;
+}