0
点赞
收藏
分享

微信扫一扫

【力扣刷题】1020.飞地的数量(python)

乱世小白 2022-02-12 阅读 17

题目简述

给你一个大小为 m x n 的二进制矩阵 grid ,其中 0 表示一个海洋单元格、1 表示一个陆地单元格。

一次 移动 是指从一个陆地单元格走到另一个相邻(上、下、左、右)的陆地单元格或跨过 grid 的边界。

返回网格中 无法 在任意次数的移动中离开网格边界的陆地单元格的数量。

示例 1:

在这里插入图片描述
输入:grid = [[0,0,0,0],[1,0,1,0],[0,1,1,0],[0,0,0,0]]
输出:3
解释:有三个 1 被 0 包围。一个 1 没有被包围,因为它在边界上。

示例 2:

在这里插入图片描述
输入:grid = [[0,1,1,0],[0,0,1,0],[0,0,1,0],[0,0,0,0]]
输出:0
解释:所有 1 都在边界上或可以到达边界。

提示

  • m == grid.length
  • n == grid[i].length
  • 1 <= m, n <= 500
  • grid[i][j] 的值为 0 或 1

题目分析

本题使用深度优先可以很好解决,从角落开始遍历,如果遇到挨着的1就改为0,最终遍历结束后统计总共还剩多少个1就是飞地的数量。代码如下:

class Solution:
    def numEnclaves(self, grid: List[List[int]]) -> int:
        def dfs(i, j):
            grid[i][j] = 0
            for x, y in [(i-1,j),(i+1,j),(i,j-1),(i,j+1)]:
                if 0 <= x < m and 0 <= y < n and grid[x][y]:
                    dfs(x, y)                    

        m, n = len(grid), len(grid[0])
        for i in range(m):
            if grid[i][0] == 1:    dfs(i, 0)
            if grid[i][n-1] == 1:  dfs(i, n-1)   
        for j in range(n):
            if grid[0][j] == 1:    dfs(0, j)
            if grid[m-1][j] == 1:  dfs(m-1, j)
                
        ans = 0
        for i in range(m):
            for j in range(n):
                if grid[i][j]: ans += 1
        return ans

举报

相关推荐

0 条评论