这里有一幅服务器分布图,服务器的位置标识在 m * n 的整数矩阵网格 grid 中,1 表示单元格上有服务器,0 表示没有。
如果两台服务器位于同一行或者同一列,我们就认为它们之间可以进行通信。
请你统计并返回能够与至少一台其他服务器进行通信的服务器的数量。
示例 1:
输入:grid = [[1,0],[0,1]]
输出:0
解释:没有一台服务器能与其他服务器进行通信。
代码
class Solution {public int countServers(int[][] grid) {int t=0;int[][] dir = new int[][]{{-1, 0}, {1, 0}, {0, 1}, {0, -1}};for(int i=0;i<grid.length;i++)for(int j=0;j<grid[0].length;j++)if(grid[i][j]==1){int temp=getServers(grid,i,j,dir);t+=temp==1?0:temp;}return t;}public int getServers(int[][] grid,int x,int y, int[][] dir) {if (x < 0 || x >= grid.length || y < 0 || y >= grid[0].length||grid[x][y]==-1||grid[x][y]==0)return 0;grid[x][y]=-1;int res=1;for(int i=1;i<= Math.max(grid.length,grid[0].length);i++)//dfs同一行同一列的for (int[] d : dir) {int nextX = i*d[0] + x, nextY = i*d[1] + y;res+=getServers(grid, nextX, nextY, dir);}return res;}
}