题目
- 图像渲染
有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i][j] 表示该图画的像素值大小。
你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[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,因为它不是在上下左右四个方向上与初始点相连的像素点。
示例 2:
输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]
提示:
m == image.length
n == image[i].length
1 <= m, n <= 50
0 <= image[i][j], newColor < 216
0 <= sr < m
0 <= sc < n
思路
该题可用广度优先搜索,也可以用深度优先搜索,这里给出的代码是广度优先搜索。
这里先判断渲染的新颜色和旧颜色是否相同,若相同,则直接返回原来的二级指针即可;
不同时,则先记录下初始坐标的值,并渲染初始坐标,然后开一个二维数组,用来记录符合要求的坐标,然后遍历所有符合要求的坐标并渲染它们即可(代码中while里的内容即完成该操作)。
代码
const int dx[4]={1,0,0,-1};
const int dy[4]={0,1,-1,0};
int** floodFill(int** image, int imageSize, int* imageColSize,
int sr, int sc, int newColor, int* returnSize, int** returnColumnSizes){
int m=imageSize,n=imageColSize[0];
*returnSize=imageSize;
*returnColumnSizes=imageColSize;
int curcolor=image[sr][sc];
if(curcolor==newColor){
return image;
}
int f[m*n][2];
int l=0,r=0;
f[r][0]=sr;f[r][1]=sc;
r++;
image[sr][sc]=newColor;
while(l<r){
int x=f[l][0],y=f[l][1];
l++;
for(int i=0;i<4;i++){
int ix=x+dx[i],iy=y+dy[i];
if(ix>=0&&ix<m&&iy>=0&&iy<n&&curcolor==image[ix][iy]){
f[r][0]=ix,f[r][1]=iy;
r++;
image[ix][iy]=newColor;
}
}
}
return image;
}