华为机考真题 -- 可以组成网络的服务器

题目描述：

在一个机房中，服务器的位置标识在 n*m 的整数矩阵网格中，1 表示单元格上有服务器，0表示没有。如果两台服务器位于同一行或者同一列中紧邻的位置，则认为它们之间可以组成一个局域网。

请你统计机房中最大的局域网包含的服务器个数。

输入描述：

第一行输入两个正整数，n 和 m，0<n,m<=100;
之后的行数为 n*m 的二维数组，代表服务器信息;

输出描述：

最大局域网包含的服务器个数;

示例 1：

输入
2 2
1 0
1 1

输出
3
说明：[0][0]、[1][0]、[1][1]三台服务器相互连接，可以组成局域网

C++源码：

#include <iostream>
#include <vector>

using namespace std;

int dx[] = { 0, 1, 0, -1 };
int dy[] = { 1, 0, -1, 0 };

bool isValid(int x, int y, int n, int m) {
    return x >= 0 && x < n && y >= 0 && y < m;
}

void dfs(vector<vector<int>>& grid, vector<vector<bool>>& visited, int x, int y, int& count) {
    visited[x][y] = true;
    ++count;

    for (int i = 0; i < 4; ++i) {
        int newX = x + dx[i];
        int newY = y + dy[i];

        if (isValid(newX, newY, grid.size(), grid[0].size()) && grid[newX][newY] == 1 && !visited[newX][newY]) {
            dfs(grid, visited, newX, newY, count);
        }
    }
}

int main() {
    int n, m;
    cin >> n >> m;

    vector<vector<int>> grid(n, vector<int>(m));
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            cin >> grid[i][j];
        }
    }

    vector<vector<bool>> visited(n, vector<bool>(m, false));
    int maxCount = 0;

    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            if (grid[i][j] == 1 && !visited[i][j]) {
                int count = 0;
                dfs(grid, visited, i, j, count);
                maxCount = max(maxCount, count);
            }
        }
    }

    cout << maxCount << endl;
    system("pause");
    return 0;
}