统计参与通信的服务器

/*
 * 思路：
 * 1. 使用Java Stream API进行行和列的遍历，并统计每行和每列的服务器数量。
 * 2. 再次遍历矩阵，使用Stream判断每个服务器是否能与其他服务器通信。
 * 3. 统计能够与至少一台其他服务器进行通信的服务器数量。
 */

import java.util.stream.IntStream;

public class Solution {
    public int countServers(int[][] grid) {
        int m = grid.length, n = grid[0].length;
        int[] rowCount = new int[m];
        int[] colCount = new int[n];

        // 计算每一行和每一列的服务器数量
        IntStream.range(0, m).forEach(i -> IntStream.range(0, n).forEach(j -> {
            if (grid[i][j] == 1) {
                rowCount[i]++;
                colCount[j]++;
            }
        }));

        // 统计能够与至少一台其他服务器进行通信的服务器数量
        return (int) IntStream.range(0, m).flatMap(i -> IntStream.range(0, n).filter(j -> grid[i][j] == 1 && (rowCount[i] > 1 || colCount[j] > 1))).count();
    }
}