Day16-Number of Islands
问题描述:
给了一个2维数组,里面是由1和0组成。规则是这样的,1代表陆地,0代表海水,让我们寻找小岛,那小岛肯定有一个特点,被水隔离,连续的陆地组成一块小岛,而本题中连续的定义为上下左右相邻(不包括对角线)。基于这样的规则让我们找寻有几个岛。
Example
Example 1:
Input:
11110
11010
11000
00000
Output: 1
Example 2:
Input:
11000
11000
00100
00011
Output: 3
解法一:
通过深度优先搜素。DFS,遇到1的时候深度遍历其左右上下相邻的点,并将为1的点标记为0,这样最后在初始for循环中用到了几次的dfs,则有几个小岛。并且遍历完整个数组全是“水”。
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
# 深度优先遍历DFS
# 如果遇到1,说明当前肯定有一个小岛了,此时将该位置设置为0,并且继续对齐上下左右位置进行DFS.则最后用了多少次辅助函数即有多少个小岛。
result = 0
for i in range(len(grid)):
for j in range(len(grid[i])):
if grid[i][j] == '1':
result += 1
self.dfs(grid,i,j)
return result
def dfs(self,grid,row,col):
temp = [[-1,0],[0,1],[0,-1],[1,0]]#这个表示当前位置前后左右移动的方向
grid[row][col] = '0'
for t in temp:
next_r,next_c = t[0] + row,t[1] + col
if next_r >=0 and next_c >= 0 and next_r < len(grid) and next_c < len(grid[row]):
if grid[next_r][next_c] == '1':
self.dfs(grid,next_r,next_c)
解法二:
广度优先搜索算法。BFS.
设置一个数组存储当前为1的元素位置,然后就是出队和入队的操作。出队一个将其上下左右的位置都访问一遍,如果为1则入队,并且在网格中将该位置设为0(已访问过)。最后队列为空且网格都为0则访问结束。
class Solution:
def numIslands(self, grid: List[List[str]]) -> int:
#BFS
if len(grid) <= 0 or len(grid[0]) <= 0:
return 0
temp = [[0,-1],[0,1],[1,0],[-1,0]]
queue = []
result = 0
for i in range(len(grid)):
for j in range(len(grid[i])):
if grid[i][j] == '1':
queue.append([i,j])
result += 1
grid[i][j] = '0'
while len(queue) > 0:
q = queue.pop()
for t in temp:
row,col = q[0] + t[0],q[1] + t[1]
if row >= 0 and col >= 0 and row < len(grid) and col < len(grid[row]):
if grid[row][col] == '1':
print(row,col)
queue.append([row,col])
grid[row][col] = '0'
return result