Question

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

English Version

题目描述

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

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

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

 

示例 1：

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

示例 2：

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

示例 3：

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

 

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m <= 250
• 1 <= n <= 250
• grid[i][j] == 0 or 1

解法

方法一：计数

我们可以统计每一行、每一列的服务器数量，然后遍历每个服务器，若当前服务器所在的行或者列的服务器数量超过 $1$，说明当前服务器满足条件，结果加 $1$。

遍历结束后，返回结果即可。

时间复杂度 $O(m \times n)$，空间复杂度 $O(m + n)$。其中 $m$ 和 $n$ 分别为矩阵的行数和列数。

class Solution:
  def countServers(self, grid: List[List[int]]) -> int:
    m, n = len(grid), len(grid[0])
    row = [0] * m
    col = [0] * n
    for i in range(m):
      for j in range(n):
        if grid[i][j]:
          row[i] += 1
          col[j] += 1
    return sum(
      grid[i][j] and (row[i] > 1 or col[j] > 1)
      for i in range(m)
      for j in range(n)
    )

class Solution {
  public int countServers(int[][] grid) {
    int m = grid.length, n = grid[0].length;
    int[] row = new int[m];
    int[] col = new int[n];
    for (int i = 0; i < m; ++i) {
      for (int j = 0; j < n; ++j) {
        if (grid[i][j] == 1) {
          row[i]++;
          col[j]++;
        }
      }
    }
    int ans = 0;
    for (int i = 0; i < m; ++i) {
      for (int j = 0; j < n; ++j) {
        if (grid[i][j] == 1 && (row[i] > 1 || col[j] > 1)) {
          ++ans;
        }
      }
    }
    return ans;
  }
}

class Solution {
public:
  int countServers(vector<vector<int>>& grid) {
    int m = grid.size(), n = grid[0].size();
    vector<int> row(m), col(n);
    for (int i = 0; i < m; ++i) {
      for (int j = 0; j < n; ++j) {
        if (grid[i][j]) {
          ++row[i];
          ++col[j];
        }
      }
    }
    int ans = 0;
    for (int i = 0; i < m; ++i) {
      for (int j = 0; j < n; ++j) {
        ans += grid[i][j] && (row[i] > 1 || col[j] > 1);
      }
    }
    return ans;
  }
};

func countServers(grid [][]int) (ans int) {
  m, n := len(grid), len(grid[0])
  row, col := make([]int, m), make([]int, n)
  for i := range grid {
    for j, x := range grid[i] {
      if x == 1 {
        row[i]++
        col[j]++
      }
    }
  }
  for i := range grid {
    for j, x := range grid[i] {
      if x == 1 && (row[i] > 1 || col[j] > 1) {
        ans++
      }
    }
  }
  return
}

function countServers(grid: number[][]): number {
  const m = grid.length;
  const n = grid[0].length;
  const row = new Array(m).fill(0);
  const col = new Array(n).fill(0);
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      if (grid[i][j] === 1) {
        row[i]++;
        col[j]++;
      }
    }
  }
  let ans = 0;
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      if (grid[i][j] === 1 && (row[i] > 1 || col[j] > 1)) {
        ans++;
      }
    }
  }
  return ans;
}