Leetcode 5272：统计参与通信的服务器

题目描述

这里有一幅服务器分布图，服务器的位置标识在 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-servers-that-communicate
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路

class Solution {
public:
    int countServers(vector<vector<int>>& grid) {
        int ans = 0;
        vector<int> dx = {-1,0,1,0};
        vector<int> dy = {0,-1,0,1};
        int m = grid.size(),n = grid[0].size();
        for(int i=0;i<m;++i){
            for(int j=0;j<n;++j){
                if(grid[i][j]==0) continue;
                int flag = 0;
                for(int k=0;k<4;++k){
                    int x = i+dx[k];
                    int y = j+dy[k];
                    while(x>=0&&x<m&&y>=0&&y<n){
                        if(grid[x][y] == 1){
                            flag = 1;
                            break;
                        }
                        x += dx[k];
                        y += dy[k];
                    }
                    if(flag == 1) break;
                }
                ans += flag;
            }
        }
        return ans;
    }
};