refactor Flood Fill

Author: za-bannai

go/flood_fill.go

@@ -3,21 +3,16 @@ package main
 
 func floodFill(image [][]int, sr int, sc int, color int) [][]int {
 	originalColor := image[sr][sc]
-
 	if originalColor == color {
 		return image
 	}
 	image[sr][sc] = color
-
-	directions := [][]int{{-1, 0}, {1, 0}, {0, -1}, {0, 1}}
-	for _, d := range directions {
-		nSr, nSc := sr+d[0], sc+d[1]
-		if 0 <= nSr && nSr < len(image) &&
-			0 <= nSc && nSc < len(image[0]) &&
-			image[nSr][nSc] == originalColor {
-			image = floodFill(image, nSr, nSc, color)
+	directions := [][]int{{0, 1}, {0, -1}, {1, 0}, {-1, 0}}
+	for _, direction := range directions {
+		newSr, newSc := sr+direction[0], sc+direction[1]
+		if 0 <= newSr && newSr < len(image) && 0 <= newSc && newSc < len(image[0]) && image[newSr][newSc] == originalColor {
+			floodFill(image, newSr, newSc, color)
 		}
 	}
-
 	return image
 }