Merge pull request #211 from gostool/leetcode1034

Leetcode1034
This commit is contained in:
halfrost
2021-12-18 21:21:21 +08:00
3 changed files with 211 additions and 0 deletions

View File

@ -0,0 +1,50 @@
package leetcode
type point struct {
x int
y int
}
type gridInfo struct {
m int
n int
grid [][]int
originalColor int
}
func colorBorder(grid [][]int, row, col, color int) [][]int {
m, n := len(grid), len(grid[0])
dirs := []point{{1, 0}, {-1, 0}, {0, 1}, {0, -1}}
vis := make([][]bool, m)
for i := range vis {
vis[i] = make([]bool, n)
}
var borders []point
gInfo := gridInfo{
m: m,
n: n,
grid: grid,
originalColor: grid[row][col],
}
dfs(row, col, gInfo, dirs, vis, &borders)
for _, p := range borders {
grid[p.x][p.y] = color
}
return grid
}
func dfs(x, y int, gInfo gridInfo, dirs []point, vis [][]bool, borders *[]point) {
vis[x][y] = true
isBorder := false
for _, dir := range dirs {
nx, ny := x+dir.x, y+dir.y
if !(0 <= nx && nx < gInfo.m && 0 <= ny && ny < gInfo.n && gInfo.grid[nx][ny] == gInfo.originalColor) {
isBorder = true
} else if !vis[nx][ny] {
dfs(nx, ny, gInfo, dirs, vis, borders)
}
}
if isBorder {
*borders = append(*borders, point{x, y})
}
}

View File

@ -0,0 +1,53 @@
package leetcode
import (
"fmt"
"testing"
)
type question1034 struct {
para1034
ans1034
}
// para 是参数
type para1034 struct {
grid [][]int
row int
col int
color int
}
// ans 是答案
type ans1034 struct {
ans [][]int
}
func Test_Problem1034(t *testing.T) {
qs := []question1034{
{
para1034{[][]int{{1, 1}, {1, 2}}, 0, 0, 3},
ans1034{[][]int{{3, 3}, {3, 2}}},
},
{
para1034{[][]int{{1, 2, 2}, {2, 3, 2}}, 0, 1, 3},
ans1034{[][]int{{1, 3, 3}, {2, 3, 3}}},
},
{
para1034{[][]int{{1, 1, 1}, {1, 1, 1}, {1, 1, 1}}, 1, 1, 2},
ans1034{[][]int{{2, 2, 2}, {2, 1, 2}, {2, 2, 2}}},
},
}
fmt.Printf("------------------------Leetcode Problem 1034------------------------\n")
for _, q := range qs {
_, p := q.ans1034, q.para1034
fmt.Printf("【input】:%v 【output】:%v\n", p, colorBorder(p.grid, p.row, p.col, p.color))
}
fmt.Printf("\n\n\n")
}

View File

@ -0,0 +1,108 @@
# [1034. Coloring A Border](https://leetcode-cn.com/problems/coloring-a-border/)
## 题目
You are given an m x n integer matrix grid, and three integers row, col, and color. Each value in the grid represents the color of the grid square at that location.
Two squares belong to the same connected component if they have the same color and are next to each other in any of the 4 directions.
The border of a connected component is all the squares in the connected component that are either 4-directionally adjacent to a square not in the component, or on the boundary of the grid (the first or last row or column).
You should color the border of the connected component that contains the square grid[row][col] with color.
Return the final grid.
**Example 1**:
Input: grid = [[1,1],[1,2]], row = 0, col = 0, color = 3
Output: [[3,3],[3,2]]
**Example 2**:
Input: grid = [[1,2,2],[2,3,2]], row = 0, col = 1, color = 3
Output: [[1,3,3],[2,3,3]]
**Example 3**:
Input: grid = [[1,1,1],[1,1,1],[1,1,1]], row = 1, col = 1, color = 2
Output: [[2,2,2],[2,1,2],[2,2,2]]
**Constraints:**
- m == grid.length
- n == grid[i].length
- 1 <= m, n <= 50
- 1 <= grid[i][j], color <= 1000
- 0 <= row < m
- 0 <= col < n
## 题目大意
给你一个大小为 m x n 的整数矩阵 grid 表示一个网格另给你三个整数 rowcol color 网格中的每个值表示该位置处的网格块的颜色
当两个网格块的颜色相同而且在四个方向中任意一个方向上相邻时它们属于同一连通分量
边界在连通分量的块中前提并且满足以下条件之一
1要么上下左右存在一个块不在连通分量里面
2要么这个块的位置在整个grid的边框上
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的连通分量的边界进行着色并返回最终的网格 grid
## 解题思路
- 用bfs进行遍历选出边界使用color给边界着色
## 代码
```go
package leetcode
type point struct {
x int
y int
}
type gridInfo struct {
m int
n int
grid [][]int
originalColor int
}
func colorBorder(grid [][]int, row, col, color int) [][]int {
m, n := len(grid), len(grid[0])
dirs := []point{{1, 0}, {-1, 0}, {0, 1}, {0, -1}}
vis := make([][]bool, m)
for i := range vis {
vis[i] = make([]bool, n)
}
var borders []point
gInfo := gridInfo{
m: m,
n: n,
grid: grid,
originalColor: grid[row][col],
}
dfs(row, col, gInfo, dirs, vis, &borders)
for _, p := range borders {
grid[p.x][p.y] = color
}
return grid
}
func dfs(x, y int, gInfo gridInfo, dirs []point, vis [][]bool, borders *[]point) {
vis[x][y] = true
isBorder := false
for _, dir := range dirs {
nx, ny := x+dir.x, y+dir.y
if !(0 <= nx && nx < gInfo.m && 0 <= ny && ny < gInfo.n && gInfo.grid[nx][ny] == gInfo.originalColor) {
isBorder = true
} else if !vis[nx][ny] {
dfs(nx, ny, gInfo, dirs, vis, borders)
}
}
if isBorder {
*borders = append(*borders, point{x, y})
}
}
```