1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
public int[][] floodFill(int[][] image, int sr, int sc, int color) { int oldColor = image[sr][sc]; dfs(image, oldColor, sr, sc, color); return image; }
public void dfs(int[][] image, int oldColor, int sr, int sc, int newColor) { int rLength = image.length; int cLength = image[0].length; if(sr < 0 || sc < 0 || sr >= rLength || sc >= cLength || image[sr][sc] != oldColor || image[sr][sc] == newColor){ return; } image[sr][sc] = newColor;
dfs(image, oldColor, sr - 1, sc, newColor); dfs(image, oldColor, sr + 1, sc, newColor); dfs(image, oldColor, sr, sc - 1, newColor); dfs(image, oldColor, sr, sc + 1, newColor); }
|