Create count-servers-that-communicate.py

Author: kamyu104

Python/count-servers-that-communicate.py

@@ -0,0 +1,22 @@
+# Time:  O(m * n)
+# Space: O(m + n)
+
+class Solution(object):
+    def countServers(self, grid):
+        """
+        :type grid: List[List[int]]
+        :rtype: int
+        """
+        rows, cols = [0]*len(grid), [0]*len(grid[0])
+        for i in xrange(len(grid)):
+            for j in xrange(len(grid[0])):
+                if grid[i][j]:
+                    rows[i] += 1
+                    cols[j] += 1
+        result = 0
+        for i in xrange(len(grid)):
+            for j in xrange(len(grid[0])):
+                if grid[i][j] and (rows[i] > 1 or cols[j] > 1):
+                    result += 1
+        return result
+