Leetcode 1396 Solution

This article provides solution to leetcode question 1396 (count-servers-that-communicate)
Categories:
Tags:
Access this page by simply typing in "lcs 1396" in your browser address bar if you have bunnylol configured.

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

Solution

class Solution:
    def countServers(self, grid: List[List[int]]) -> int:
        m = len(grid)
        n = len(grid[0])

        row_srvs = [0] * m
        col_srvs = [0] * n

        for i in range(m):
            s = 0
            for j in range(n):
                s += grid[i][j]
            row_srvs[i] = s

        for j in range(n):
            s = 0
            for i in range(m):
                s += grid[i][j]
            col_srvs[j] = s

        ans = 0
        for i in range(m):
            for j in range(n):
                if grid[i][j] == 0:
                    continue

                if row_srvs[i] > 1 or col_srvs[j] > 1:
                    ans += 1
        return ans