leetcode_1267 统计参与通信的服务器

182 阅读1分钟

要求

这里有一幅服务器分布图,服务器的位置标识在 m * n 的整数矩阵网格 grid 中,1 表示单元格上有服务器,0 表示没有。

如果两台服务器位于同一行或者同一列,我们就认为它们之间可以进行通信。

请你统计并返回能够与至少一台其他服务器进行通信的服务器的数量。

示例 1: image.png

输入:grid = [[1,0],[0,1]]
输出:0
解释:没有一台服务器能与其他服务器进行通信。

示例 2:

image.png

输入:grid = [[1,0],[1,1]]
输出:3
解释:所有这些服务器都至少可以与一台别的服务器进行通信。

示例 3:

image.png

输入:grid = [[1,1,0,0],[0,0,1,0],[0,0,1,0],[0,0,0,1]]
输出:4
解释:第一行的两台服务器互相通信,第三列的两台服务器互相通信,但右下角的服务器无法与其他服务器通信。

代码详解

class Solution:
    def countServers(self, grid: List[List[int]]) -> int:
        if not grid:
            return 0
        
        row_count = [0] * len(grid)
        col_count = [0] * len(grid[0])
        for row in range(len(grid)):
            for col in range(len(grid[0])):
                if grid[row][col] == 1:
                    row_count[row] += 1
                    col_count[col]  += 1
        
        result = 0
        for row in range(len(grid)):
            for col in range(len(grid[0])):
                if grid[row][col] == 1 and (row_count[row] > 1 or col_count[col] > 1):
                    result += 1
        return result

image.png

解题思路:我们先把行、列的总的电脑数量进行一个统计,然后我们遍历整个数据,当位置上有电脑的时候,他的行 or 列上电脑总数超过了1台,说明他一定可以完成通信。