zl程序教程

您现在的位置是:首页 >  其他

当前栏目

leetcode 463. Island Perimeter

2023-09-14 09:11:53 时间

You are given a map in form of a two-dimensional integer grid where 1 represents land and 0 represents water. Grid cells are connected horizontally/vertically (not diagonally). The grid is completely surrounded by water, and there is exactly one island (i.e., one or more connected land cells). The island doesn't have "lakes" (water inside that isn't connected to the water around the island). One cell is a square with side length 1. The grid is rectangular, width and height don't exceed 100. Determine the perimeter of the island.

Example:

[[0,1,0,0],
 [1,1,1,0],
 [0,1,0,0],
 [1,1,0,0]]

Answer: 16
Explanation: The perimeter is the 16 yellow stripes in the image below:

 

我的解法:就是数数而已

class Solution(object):
    def islandPerimeter(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        # for each element in island, calculate stripes and sum them
        row = len(grid)
        col = len(grid[0])
        ans = 0
        for i in range(0, row):
            for j in range(0, col):
                if grid[i][j] == 1:
                    if i==0 or grid[i-1][j] == 0:
                        ans += 1                    
                    if i==row-1 or grid[i+1][j] == 0:
                        ans += 1
                    if j==0 or grid[i][j-1] == 0:
                        ans += 1
                    if j==col-1 or grid[i][j+1]==0:
                        ans += 1                        
        return ans

更简单的做法,统计为1的矩形个数x,以及上下和左右相邻的边数y,结果为4x-2y

class Solution(object):
    def islandPerimeter(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        # for each element in island, calculate stripes and sum them
        row = len(grid)
        col = len(grid[0])
        ans = 0
        for i in range(0, row):
            for j in range(0, col):
                if grid[i][j] == 1:
                    ans += 4
                    if i>0 and grid[i-1][j] == 1:
                        ans -= 2
                    if j>0 and grid[i][j-1] == 1:
                        ans -= 2                        
        return ans

还有一种DFS版本,

class Solution(object):
    def islandPerimeter(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        # for each element in island, calculate stripes and sum them
        row = len(grid)
        col = len(grid[0])
        self.ans = 0
        
        def dfs(grid,i,j):
            if i<0 or i>=row or j<0 or j>=col:
                return
            if grid[i][j] == 0: return
            if grid[i][j] == 1:
                grid[i][j] = -1
                if i==0 or grid[i-1][j] == 0:
                    self.ans += 1                    
                if i==row-1 or grid[i+1][j] == 0:
                    self.ans += 1
                if j==0 or grid[i][j-1] == 0:
                    self.ans += 1
                if j==col-1 or grid[i][j+1]==0:
                    self.ans += 1
                dfs(grid, i-1, j)
                dfs(grid, i+1, j)
                dfs(grid, i, j-1)
                dfs(grid, i, j+1)
        
        for i in range(0, row):
            for j in range(0, col):
                if grid[i][j] == 1:
                    dfs(grid, i, j)                       
        return self.ans

DFS的代码结构:

dfs(xxx):

    process(xxx.val)

    dfs(xxx.left)

    dfs(xxx.right)

也就和树的深度优先,先序遍历类似!