华为OD 2024笔试机试 - 可以组成网络的服务器 (python C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

在一个机房中，服务器的位置标识在 n*m 的整数矩阵网格中，1 表示单元格上有服务器，0 表示没有。如果两台服务器位于同一行或者同一列中紧邻的位置，则认为它们之间可以组成一个局域网。

请你统计机房中最大的局域网包含的服务器个数。

输入描述

第一行输入两个正整数，n和m，0<n,m<=100

之后为n*m的二维数组，代表服务器信息

输出描述

最大局域网包含的服务器个数。

用例

输入

2 2
1 0
1 1

输出

3

说明 [0][0]、[1][0]、[1][1]三台服务器相互连接，可以组成局域网

解题思路

可以用广度优先搜索（BFS）来解决。我们需要遍历整个矩阵，对于每一个发现的服务器（即矩阵中的 1），我们使用 BFS 来统计和其相连的所有服务器的数量，记录最大的局域网包含的服务器个数。

1. 遍历整个矩阵，寻找每个服务器（即矩阵中的 1）。
2. 对每个发现的服务器，使用 BFS 来找到所有和它直接或间接相连的服务器，并计算这些服务器的数量。
3. 在 BFS 中，我们使用队列来存储需要访问的服务器，同时标记已访问的服务器，避免重复计算。
4. 在遍历过程中，记录最大的局域网包含的服务器个数。
5. 输出最大的局域网包含的服务器个数。

参考代码

# 获取输入
n, m = map(int, input().split())
# 构建矩阵
matrix = [list(map(int, input().split())) for _ in range(n)]

# 方向数组，上下左右四个方向的偏移量
offsets = ((-1, 0), (1, 0), (0, -1), (0, 1))

# 广度优先搜索（BFS）函数，用于查找连通区域的大小
def bfs(i, j):
    count = 1
    matrix[i][j] = 0  # 将访问过的位置标记为0，防止重复访问
    
    queue = [[i, j]]  # 初始化队列，存储待访问的节点
    
    while queue:
        x, y = queue.pop(0)  # 弹出队列中的第一个节点
        
        # 遍历四个方向
        for offsetX, offsetY in offsets:
            newX = x + offsetX
            newY = y + offsetY
            
            # 判断新位置是否在矩阵范围内且未被访问
            if 0 <= newX < n and 0 <= newY < m and matrix[newX][newY] == 1:
                count += 1
                matrix[newX][newY] = 0  # 标记新位置已访问
                queue.append([newX, newY])  # 将新位置加入队列
    
    return count

max_area = 0  # 记录最大连通区域的大小

# 遍历矩阵中的每一个位置
for i in range(n):
    for j in range(m):
        if matrix[i][j] == 1:  # 如果位置为1，说明是新的连通区域的起点
            max_area = max(max_area, bfs(i, j))  # 更新最大连通区域的大小

print(max_area)