算法练习:统计参与通信的服务器

题目描述:

这里有一幅服务器分布图，服务器的位置标识在 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
解释：第一行的两台服务器互相通信，第三列的两台服务器互相通信，但右下角的服务器无法与其他服务器通信。

解题思路:

刚读这个题的时候脑子中浮现出很多的方法,map\set…好像总是怎么牛×就会怎样想,结果看了一下官方的解题思路,我哭啦,居然就是直接遍历然后记录

1. 遍历二维数组,如果某个位置有服务器,则记录对应行和对应列的数组的对应位置就加一来记录
2. 再遍历一遍二维数组,如果当前位置对应的行或者列的服务器数大于1,则该服务器就会与其他服务进行沟通,结果加一

解题代码:

 public static int countServers(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;

        int[] rowCount = new int[m];
        int[] colCount = new int[n];

        for (int i = 0; i <m ; i++){
            for(int j = 0; j <n ; j++){
                if(grid[i][j] ==1){
                    rowCount[i]++;
                    colCount[j]++;
                }
            }
        }
        int res=0;
        for (int i = 0; i <m ; i++){
            for(int j = 0; j <n ; j++){
                if(grid[i][j] ==1&&(rowCount[i]>1 || colCount[j]>1)){
                   res++;
                }
            }
        }

        return res ;

    }