Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Solution {
- public:
- void dfs(vector<vector<int>>& image, int i, int j, int initialColor, int color, int &n, int &m){
- if(i<0 || i>=n || j<0 || j>=m)return;
- if(image[i][j]!=initialColor)return;
- image[i][j] = color;
- dfs(image, i-1,j,initialColor,color, n,m);
- dfs(image, i+1,j,initialColor,color, n,m);
- dfs(image, i,j-1,initialColor,color, n,m);
- dfs(image, i,j+1,initialColor,color, n,m);
- return;
- }
- vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
- int n= image.size();
- int m = image[0].size();
- int initialColor = image[sr][sc];
- if(initialColor==color)return image;
- dfs(image, sr,sc,initialColor,color, n,m);
- return image;
- }
- };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement