Fix : Floodfill infinite recursion due to same color (TheAlgorithms#4359)

Manan-09 · web-flow · commit a88abb7ac238 · 2023-09-09T20:07:59.000+02:00
Fix : Floodfill infinite recursion due to same color
diff --git a/src/main/java/com/thealgorithms/backtracking/FloodFill.java b/src/main/java/com/thealgorithms/backtracking/FloodFill.java
@@ -39,6 +39,7 @@ public static void putPixel(int[][] image, int x, int y, int newColor) {
      * @param oldColor The old color which is to be replaced in the image
      */
     public static void floodFill(int[][] image, int x, int y, int newColor, int oldColor) {
+        if (newColor == oldColor) return;
         if (x < 0 || x >= image.length) return;
         if (y < 0 || y >= image[x].length) return;
         if (getPixel(image, x, y) != oldColor) return;
diff --git a/src/test/java/com/thealgorithms/backtracking/FloodFillTest.java b/src/test/java/com/thealgorithms/backtracking/FloodFillTest.java
@@ -93,4 +93,14 @@ void testForImageThree() {
         FloodFill.floodFill(image, 0, 1, 4, 1);
         assertArrayEquals(expected, image);
     }
+
+    @Test
+    void testForSameNewAndOldColor() {
+        int[][] image = {{1, 1, 2}, {1, 0, 0}, {1, 1, 1}};
+
+        int[][] expected = {{1, 1, 2}, {1, 0, 0}, {1, 1, 1}};
+
+        FloodFill.floodFill(image, 0, 1, 1, 1);
+        assertArrayEquals(expected, image);
+    }
 }
