Skip to content

Latest commit

 

History

History
71 lines (58 loc) · 2.4 KB

README.md

File metadata and controls

71 lines (58 loc) · 2.4 KB

题目描述: 图像渲染

有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 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]内。

解题思路:

dfs,与前几天的被围绕的区域思路很类似。

时间复杂度:

O(mn)

空间复杂度

O(mn)

代码

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
        int m = image.size();
        int n = image[0].size();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (i == sr && j == sc) {
                    int k = image[i][j];
                    dfs(image, i, j, k, newColor);
                    return image;
                }
            }
        }
        return {};
    }

    void dfs (vector<vector<int>> & image, int i, int j, int k, int newColor) {
        if (i <0 || i >= image.size() || j <0 || j >= image[0].size () || image[i][j] == newColor  ||image[i][j] != k) {
            return;
        } 
        image[i][j] = newColor;
        dfs(image, i-1, j, k, newColor);
        dfs(image, i+1, j, k, newColor);
        dfs(image, i, j-1, k, newColor);
        dfs(image, i, j+1, k, newColor);
    }
};