探索 隊列和棧 圖像渲染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,
因爲它不是在上下左右四個方向上與初始點相連的像素點。
注意:
1.image 和 image[0] 的長度在範圍 [1, 50] 內。
2.給出的初始點將滿足 0 <= sr < image.length 和 0 <= sc < image[0].length。
3.image[i][j] 和 newColor 表示的顏色值在範圍 [0, 65535]內。

思路:使用dfs解法進行深度優先搜索即可

class Solution {
public:
    vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
    int color = image[sr][sc];//目標點原有的顏色
    if(newColor == color)//目標點原有的顏色和新的顏色相同
    {}
    else
    dfs(image,sr,sc,newColor,color);    
    return image;
    }
    void dfs(vector<vector<int>>& image, int sr, int sc, int newColor,int color){
        int hang=image.size();
        int lie=image[0].size();
        image[sr][sc]=newColor;
        if(sr-1>=0 && image[sr-1][sc]==color){
            dfs(image,sr-1,sc,newColor,color);
        }
        if(sr+1<hang && image[sr+1][sc]==color){
            dfs(image,sr+1,sc,newColor,color);
        }
        if(sc-1>=0 && image[sr][sc-1]==color){
            dfs(image,sr,sc-1,newColor,color);
        }
        if(sc+1<lie && image[sr][sc+1]==color){
            dfs(image,sr,sc+1,newColor,color);
        }
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章