每日一题---1034. 边界着色[力扣][Go]

题目描述

给你一个大小为 m x n 的整数矩阵 grid ,表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。
当两个网格块的颜色相同,而且在四个方向中任意一个方向上相邻时,它们属于同一 连通分量 。
连通分量的边界 是指连通分量中的所有与不在分量中的网格块相邻(四个方向上)的所有网格块,或者在网格的边界上(第一行/列或最后一行/列)的所有网格块。
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的 连通分量的边界 进行着色,并返回最终的网格 grid 。

解题代码

  1. 深度搜索
// 定义方向
var dirs = []struct{x,y int}{{0,1},{1,0},{-1,0},{0,-1}}
func colorBorder(grid [][]int, row int, col int, color int) [][]int {
	m,n := len(grid),len(grid[0])
	type point struct{ x, y int }
	//borders 用来存边界的横坐标和纵坐标
	borders := []point{}
	//initialColor 用来记录初始颜色
	initialColor := grid[row][col]
	//visit 记录是否被访问过
	visit := make([][]bool,m)
	for i, _ := range visit {
		visit[i] = make([]bool,n)
	}
	//深度搜索函数
	var dfs func(int,int)
	dfs = func(x,y int) {
		visit[x][y] = true
		isBorders := false
		for _, dir := range dirs {
			nx,ny := x+dir.x,y+dir.y
			// 判断边界条件
			if !(nx < m && nx >= 0 && ny < n && ny  >= 0 && grid[nx][ny] == initialColor) {
				isBorders = true
			} else if !visit[nx][ny] {
				visit[nx][ny] = true
				dfs(nx,ny)
			}
		}
		if isBorders {
			borders = append(borders, point{x,y})
		}
	}
	dfs(row, col)
	// 将是边界的点涂上颜色
	for _, border := range borders {
		grid[border.x][border.y] = color
	}
	return grid
}
  1. 广度搜索
// 定义方向
var dirs = []struct{x,y int}{{0,1},{1,0},{-1,0},{0,-1}}
func colorBorder(grid [][]int, row int, col int, color int) [][]int {
	m,n := len(grid),len(grid[0])
	type point struct{ x, y int }
	//borders 用来存边界的横坐标和纵坐标
	borders := []point{}
	//initialColor 用来记录初始颜色
	initialColor := grid[row][col]
	//visit 记录是否被访问过
	visit := make([][]bool,m)
	for i, _ := range visit {
		visit[i] = make([]bool,n)
	}
	// 广度搜素,使用队列实现
	q := []point{{row,col}}
	visit[row][col] = true
	for len(q) > 0 {
		p := q[0]
		q = q[1:]
		isBorders := false
		for _, dir := range dirs {
			nx,ny := dir.x + p.x,dir.y + p.y
			if !(nx < m && nx >= 0 && ny < n && ny >=0 && grid[nx][ny] == initialColor) {
				isBorders = true
			} else  if !visit[nx][ny] {
				visit[nx][ny] = true
				q = append(q, point{nx,ny})
			}
		}
		if isBorders {
			borders = append(borders, point{p.x,p.y})
		}
	}

	// 将是边界的点涂上颜色
	for _, border := range borders {
		grid[border.x][border.y] = color
	}
	return grid
}

我们数据结构课上讲到了深度搜索和广度搜索,但是没有用过,正好今天力扣这一题有深刻的让我认识到
在这里插入图片描述

提交结果

两次运行结果都为:
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值