岛的个数问题

最新推荐文章于 2022-05-10 22:20:35 发布

metabit

最新推荐文章于 2022-05-10 22:20:35 发布

阅读量697

点赞数

分类专栏：数据结构与算法文章标签： 01岛

本文链接：https://blog.csdn.net/dawnto/article/details/121662501

版权

数据结构与算法专栏收录该内容

185 篇文章 5 订阅

订阅专栏

这篇博客介绍了一个用于计算二维矩阵中1形成的岛屿数量的算法。代码中定义了`countIsLand`函数，通过遍历矩阵并使用深度优先搜索（DFS）策略来标记已访问过的1，从而避免重复计数。`infect`函数则负责递归地感染相邻的1。测试用例展示了算法的正确性。

摘要由CSDN通过智能技术生成

/*
岛问题
一个只有0和1两种数字的二维矩阵中，
上下左右能连成一片的1，算一个岛
返回矩阵中，一共有几个岛
*/



func countIsLand(m [][]int) int {
	if m == nil || m[0] == nil {
		return 0
	}
	N := len(m)
	M := len(m[0])
	res := 0
	for i := 0; i < N; i++ {
		for j := 0; j < M; j++ {
		   if m[i][j] == 1 {
			   res++
			   infect(m,i,j,N,M)
		   }
		}
	}
	return res
}

func infect(m [][]int, i, j, N, M int)  {
	if i < 0 || i >= N || j < 0 || j >= M || m[i][j] != 1 {
		return
	}
	m[i][j] = 2
	infect(m,i+1,j, N,M)
	infect(m,i-1,j, N,M)
	infect(m,i,j+1, N,M)
	infect(m,i,j-1, N,M)
}

func TestCountIsLand(t *testing.T)  { // O(N*M)
	mp := [][]int{
		{1,1,1,1,0,0,1,1,1,1},
		{1,1,1,0,1,0,1,1,1,0},
		{0,1,1,1,1,0,1,1,0,1},
		{0,0,0,0,0,1,0,0,1,1},
		{1,0,0,1,1,0,1,0,1,1},
		{1,0,1,1,0,0,1,1,1,1},
	}
	t.Log(countIsLand(mp))
}