733. 图像渲染(C++)---DFS解题

题目详情

有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。

给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。

为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。

最后返回经过上色渲染后的图像。

示例 1:

输入: 
image = [[1,1,1],[1,1,0],[1,0,1]]
sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析: 
在图像的正中间，(坐标(sr,sc)=(1,1)),
在路径上所有符合条件的像素点的颜色都被更改成2。
注意，右下角的像素没有更改为2，
因为它不是在上下左右四个方向上与初始点相连的像素点。


注意:

• image 和 image[0] 的长度在范围 [1, 50] 内。
• 给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
• image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

 

 

——题目难度:简单


 

---

-解题代码（使用flag数组来记录像素点是否已经被改色）

class Solution {
private:
	int rows, cols;
	bool flag[50][50]; //记录像素点是否已经被改色 
	
public:
	void dfs(vector>& image, int sr, int sc, int color, int newColor) {
		image[sr][sc] = newColor;
		flag[sr][sc] = true;
		
		if (sr + 1 >= 0 && sr + 1 < rows && sc >= 0 && sr < cols && 
		image[sr + 1][sc] == color && flag[sr + 1][sc] == false) dfs(image, sr + 1, sc, color, newColor);
		
		if (sr - 1 >= 0 && sr - 1 < rows && sc >= 0 && sr < cols && 
		image[sr - 1][sc] == color && flag[sr - 1][sc] == false) dfs(image, sr - 1, sc, color, newColor);	
		
		if (sr >= 0 && sr < rows && sc + 1 >= 0 && sc + 1 < cols && 
		image[sr][sc + 1] == color && flag[sr][sc + 1] == false) dfs(image, sr, sc + 1, color, newColor);
		
		if (sr >= 0 && sr < rows && sc - 1 >= 0 && sr - 1 < cols && 
		image[sr][sc - 1] == color && flag[sr][sc - 1] == false) dfs(image, sr, sc - 1, color, newColor);
	}
	
    vector> floodFill(vector>& image, int sr, int sc, int newColor) {
    	rows = image.size(), cols = image[0].size();
    	for(int i = 0; i < 50; i++) {
    		for(int j = 0; j < 50; j++) {
    			flag[i][j] = false;
    		}
    	}
		dfs(image, sr, sc, image[sr][sc], newColor);
		return image;
    }
};

 

 

 

 

-解题代码（发现如果 image[sr][sc] == newColor 就可以直接返回 image）

class Solution {
private:
	int rows, cols;
	
public:
	void dfs(vector>& image, int sr, int sc, int color, int newColor) {
		image[sr][sc] = newColor;
		
		if (sr + 1 >= 0 && sr + 1 < rows && sc >= 0 && sr < cols && 
		image[sr + 1][sc] == color) dfs(image, sr + 1, sc, color, newColor);
		
		if (sr - 1 >= 0 && sr - 1 < rows && sc >= 0 && sr < cols && 
		image[sr - 1][sc] == color) dfs(image, sr - 1, sc, color, newColor);	
		
		if (sr >= 0 && sr < rows && sc + 1 >= 0 && sc + 1 < cols && 
		image[sr][sc + 1] == color) dfs(image, sr, sc + 1, color, newColor);
		
		if (sr >= 0 && sr < rows && sc - 1 >= 0 && sr - 1 < cols && 
		image[sr][sc - 1] == color) dfs(image, sr, sc - 1, color, newColor);
	}
	
    vector> floodFill(vector>& image, int sr, int sc, int newColor) {
    	if (image[sr][sc] == newColor) return image;
    	rows = image.size(), cols = image[0].size();
		dfs(image, sr, sc, image[sr][sc], newColor);
		return image;
    }
};