遍历二维数组并列出连通的1的位置。
创始人
2024-12-03 11:01:16
0次

可以使用深度优先搜索进行解决。我们遍历整个二维数组,当遇到值为1的格子时,就从该格子开始进行深度优先搜索,将所有连通的1标记。然后跳到下一个未被标记的格子,继续进行搜索操作。

下面是示例代码:

def dfs(grid, r, c):
    # 判断是否越界或者该位置已经搜索过
    if r < 0 or c < 0 or r >= len(grid) or c >= len(grid[0]) or grid[r][c] != 1:
        return
    
    # 标记该位置被搜索过
    grid[r][c] = -1
    
    # 继续搜索该位置周围的四个格子
    dfs(grid, r + 1, c)
    dfs(grid, r - 1, c)
    dfs(grid, r, c + 1)
    dfs(grid, r, c - 1)
    

def connected_ones(grid):
    # 初始化结果列表
    result = []
    
    # 遍历整个二维数组
    for r in range(len(grid)):
        for c in range(len(grid[0])):
            # 如果该位置是1且没有被标记过,则开始搜索
            if grid[r][c] == 1:
                dfs(grid, r, c)
                # 将搜索到的结果添加到结果列表中
                result.append((r, c))
    
    # 返回结果
    return result

使用方法如下:

grid = [
    [0, 1, 0, 1],
    [1, 1, 0, 0],
    [0, 0, 1, 0],
    [0, 1, 0, 0]
]

result = connected_ones(grid)

print(result) # [(0, 1), (1, 0), (1, 1), (3, 1)]

以上代码输出结果为:

[(0,

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...