LeetCode1034。ボーダーカラーリング(BFS / DFS)

記事ディレクトリ

1.タイトル

2次元の整数グリッドが与えられた場合、グリッド内の各値はその位置のグリッドブロックの色を表します。

2つのグリッドブロックが同じ色で、4方向のいずれかに隣接している場合のみ、それらは同じ接続されたコンポーネントに属します。

接続されたコンポーネントの境界は、コンポーネントではない正方形に隣接している接続されたコンポーネント内のすべての正方形(4方向)、またはグリッドの境界上のすべての正方形(最初の行/列または最後の行/列)を指します。平方。

(r0、c0)にあるグリッドブロックと色colorを指定し、指定された色colorを使用して、指定されたグリッドブロックの接続されたコンポーネントの境界に色を付け、最終的なグリッドグリッドを返します。

示例 1:
输入:grid = [[1,1],[1,2]], 
r0 = 0, c0 = 0, color = 3
输出:[[3, 3], [3, 2]]

示例 2:
输入:grid = [[1,2,2],[2,3,2]], 
r0 = 0, c0 = 1, color = 3
输出:[[1, 3, 3], [2, 3, 3]]

示例 3:
输入:grid = [[1,1,1],[1,1,1],[1,1,1]], 
r0 = 1, c0 = 1, color = 2
输出:[[2, 2, 2], [2, 1, 2], [2, 2, 2]]
 
提示:
1 <= grid.length <= 50
1 <= grid[0].length <= 50
1 <= grid[i][j] <= 1000
0 <= r0 < grid.length
0 <= c0 < grid[0].length
1 <= color <= 1000

ソース:LeetCodeリンク:https://leetcode-cn.com/problems/coloring-a-border
著作権はLeetCode が所有しています商用転載については、正式な許可書にご連絡ください。非商用転載については、出典を明記してください。

2.問題解決

  • シンプルなBFS / DFSで十分
  • 判定条件:周囲のポイントが範囲外の場合、または周囲のポイントの色が異なる場合、現在のポイントを境界としてマーク

2.1 BFS

class Solution {
    
    
public:
    vector<vector<int>> colorBorder(vector<vector<int>>& grid, int r0, int c0, int color) {
    
    
    	int m = grid.size(), n = grid[0].size(), i, j, k, x, y;
    	vector<vector<int>> dir = {
    
    {
    
    1,0},{
    
    0,1},{
    
    0,-1},{
    
    -1,0}};
    	int origin = grid[r0][c0];
    	vector<vector<bool>> vis(m, vector<bool>(n, false));
    	queue<pair<int,int>> q;
    	q.push(make_pair(r0, c0));
    	vis[r0][c0] = true;
    	while(!q.empty())
    	{
    
    
    		i = q.front().first;
    		j = q.front().second;
            q.pop();
    		for(k = 0; k < 4; ++k)
    		{
    
    
    			x = i + dir[k][0];
    			y = j + dir[k][1];
    			if(x >=0 && x<m && y>=0 && y<n)//在界内
    			{
    
    
                    if(vis[x][y]) continue;//访问过了,下一个
    				if(grid[x][y] != origin)//没有访问,颜色不同
    					grid[i][j] = color;// i, j 旁边的 x, y跟它不一样,边界
    				else//没有访问,颜色一样,正常入队
    				{
    
    
                        q.push({
    
    x,y});
                        vis[x][y] = true;
                    }
    			}
    			else//出界了
    				grid[i][j] = color;//i,j 是边界
    		}
    	}
    	return grid;
    }
};

40 ms 13 MB

2.2 DFS

class Solution {
    
    
    vector<vector<int>> dir = {
    
    {
    
    1,0},{
    
    0,1},{
    
    0,-1},{
    
    -1,0}};
    int m, n, origin, col;
public:
    vector<vector<int>> colorBorder(vector<vector<int>>& grid, int r0, int c0, int color) {
    
    
        m = grid.size(), n = grid[0].size();
        origin = grid[r0][c0], col = color;
        vector<vector<bool>> vis(m, vector<bool>(n, false));
        vis[r0][c0] = true;
        dfs(grid,r0,c0,vis);
        return grid;
    }
    void dfs(vector<vector<int>>& grid, int i, int j, vector<vector<bool>>& vis)
    {
    
    
        int x, y, k;
        for(k = 0; k < 4; ++k)
        {
    
    
            x = i + dir[k][0];
            y = j + dir[k][1];
            if(x >=0 && x<m && y>=0 && y<n)//在界内
            {
    
    
                if(vis[x][y]) continue;//访问过了,下一个
                if(grid[x][y] != origin)//没有访问,颜色不同
                    grid[i][j] = col;// i, j 旁边的 x, y跟它不一样,边界
                else//没有访问,颜色一样
                {
    
    
                    vis[x][y] = true;
                    dfs(grid, x, y, vis);
                }
            }
            else//出界了
                grid[i][j] = col;//i,j 是边界
        }
    }
};

36 ms 12.9 MB


私のCSDN ブログアドレスhttps://michael.blog.csdn.net/

QRコードを長押しまたはスキャンして、私の公式アカウント(Michael Amin)をフォローし、一緒に来て、学び、一緒に進歩してください!
マイケル・アミン

おすすめ

転載: blog.csdn.net/qq_21201267/article/details/108521257