华为机考真题 -- 图像物体的边界

题目描述：

给定一个二维数组M行N列，二维数组里的数字代表图片的像素，为了简化问题，仅包含像素1和5两种像素，每种像素代表一个物体，2个物体相邻的格子为边界，求像素1代表的物体的边界个数。像素1代表的物体的边界指与像素5相邻的像素1的格子，边界相邻的属于同一个边界，相邻需要考虑8个方向（上，下，左，右，左上，左下，右上，右下）。

其他约束

地图规格约束为：

0<M<100
0<N<100

1）如下图，与像素5的格子相邻的像素1的格子（0,0）、（0,1）、（0,2）、（1,0）、（1,2）、（2,0）、（2,1）、（2,2）、（4,4）、（4,5）、（5,4）为边界，另（0,0）、（0,1）、（0,2）、（1,0）、（1,2）、（2,0）、（2,1）、（2,2）相邻，为1个边界，（4,4）、（4,5）、（5,4）相邻，为1个边界，所以下图边界个数为2。

2）如下图，与像素5的格子相邻的像素1的格子（0,0）、（0,1）、（0,2）、（1,0）、（1,2）、（2,0）、（2,1）、（2,2）、（3,3）、（3,4）、（3,5）、（4,3）、（4,5）、（5,3）、（5,4）、（5,5）为边界，另这些边界相邻，所以下图边界个数为1。
注：（2,2）、（3,3）相邻。

输入描述：

第一行，行数M，列数N

第二行开始，是M行N列的像素的二维数组，仅包含像素1和5

输出描述：

像素1代表的物体的边界个数。

如果没有边界输出0（比如只存在像素1，或者只存在像素5）。

示例1：

输入    
6 6
1 1 1 1 1 1
1 5 1 1 1 1
1 1 1 1 1 1
1 1 1 1 1 1
1 1 1 1 1 1
1 1 1 1 1 5

输出    
2

示例2：

输入    
6 6
1 1 1 1 1 1
1 5 1 1 1 1
1 1 1 1 1 1
1 1 1 1 1 1
1 1 1 1 5 1
1 1 1 1 1 1

输出    
1

C++源码：

#include <iostream>
#include <vector>
#include <cstdlib>

using namespace std;

void dfs(int x, int y, vector<vector<bool>>& wall, vector<vector<bool>>& visited, int n, int m) {
    visited[x][y] = true;
    vector<pair<int, int>> directions = { {0, 1}, {1, 0}, {0, -1}, {-1, 0}, {1, 1}, {1, -1}, {-1, -1}, {-1, 1} };
    for (const auto& direction : directions) {
        int dx = direction.first;
        int dy = direction.second;
        int nx = x + dx, ny = y + dy;
        if (0 <= nx && nx < n && 0 <= ny && ny < m && wall[nx][ny] && !visited[nx][ny]) {
            dfs(nx, ny, wall, visited, n, m);
        }
    }
}

int main() {
    //cin.tie(nullptr); // 可加可不加
    //ios::sync_with_stdio(false);

    int N, M;
    cin >> N >> M;

    vector<vector<int>> a(N, vector<int>(M));
    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < M; ++j) {
            cin >> a[i][j];
        }
    }

    vector<vector<bool>> wall(n, vector<bool>(m, false));
    vector<vector<bool>> visited(n, vector<bool>(m, false));

    // 标记边界
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            if (a[i][j] == 5) {
                vector<pair<int, int>> directions = { {0, 1}, {1, 0}, {0, -1}, {-1, 0}, {1, 1}, {1, -1}, {-1, -1}, {-1, 1} };
                for (const auto& direction : directions) {
                    int dx = direction.first;
                    int dy = direction.second;
                    int x = i + dx, y = j + dy;
                    if (0 <= x && x < n && 0 <= y && y < m && a[x][y] != 5) {
                        wall[x][y] = true;
                    }
                }
            }
        }
    }

    // 计算并输出边界数量
    int count = 0;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            if (wall[i][j] && !visited[i][j]) {
                ++count;
                dfs(i, j, wall, visited, n, m);
            }
        }
    }
    cout << count << endl;

    system("pause");
    return 0;
}