1267. 统计参与通信的服务器

1267. 统计参与通信的服务器

题目描述：

https://leetcode-cn.com/problems/count-servers-that-communicate/

解题：计数

class Solution:
    def countServers(self, grid: List[List[int]]) -> int:
    	# 定义变量
        m, n = len(grid), len(grid[0])
        count_m, count_n = [0] * m, [0] * n
        # 统计每行，每列有多少服务器
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 1:
                    count_m[i] += 1
                    count_n[j] += 1
        # 遍历所有服务器，进行判断
        ans = 0
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 1 and (count_m[i] > 1 or count_n[j] > 1):
                    ans += 1
        return ans