图像物体的边界

alt

题目描述

给定一个二维数组M行N列,二维数组里的数字代表图片的像素,为了简化问题,仅包含像素1和5两种像素,每种像素代表一个物体2个物体相邻的格子为边界,求像素1代表的物体的边界个数。

像素1代表的物体的边界指与像素5相邻的像素1的格子,边界相邻的属于同一个边界,相邻需要考虑8个方向(上,下,左,右,左上,左下,右上,右下)。

其他约束:地图规格约束为:

  • 0<M<100
  • 0<N<100

alt

输入描述

第一行,行数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

题解

解题思路

这是一个图的深度优先搜索(DFS)问题。题目要求统计像素1代表的物体的边界个数,而像素1的边界是与像素5相邻的位置。因此,我们可以首先预先处理一下像素5的位置,将与之相邻的像素1标记为可能是边界的位置(用数字0表示)。然后,利用深度优先搜索(DFS)来统计与边界相连的区域的数量,每一轮搜索即为一组边界。

具体步骤如下:

  1. 读取输入,包括行数 rows、列数 cols 以及像素的二维数组 grid
  2. 遍历 grid,当遇到像素值为5时,预先标记其周围8个位置为0,表示可能是边界。这里可以使用两层嵌套循环遍历周围的位置,并将对应位置的像素值标记为0。
  3. 使用深度优先搜索(DFS)来统计与边界相连的区域数量。定义一个函数 dfs,在该函数中,首先检查当前坐标是否有效且对应位置的像素值为0。如果满足条件,则将当前位置的像素值标记为1,并递归地调用 dfs 函数,搜索其周围的8个位置。
  4. 主循环遍历整个二维数组 grid,对每个像素值为0的位置调用 dfs 函数,统计边界的数量。
  5. 输出最终结果。

C++

#include <bits/stdc++.h>
using namespace std;

int rows, cols;

// 验证坐标有效性
bool valid(int r, int c)
{
    return 0 <= r && r < rows && 0 <= c && c < cols;
}

// 深度优先搜索,标记为 0 的位置就是边界,之后搜索其周围的 0 的位置,并标记为 1 。
// 将所有相关联的都标记为 1
void dfs(vector<vector<int>>& grid, int r, int c)
{
    if (!valid(r, c) || grid[r][c] != 0) return;
    grid[r][c] = 1;

    for (int dr = -1; dr < 2; dr++) {
        for (int dc = -1; dc < 2; dc++) {
            int nr = r + dr, nc = c + dc;
            dfs(grid, nr, nc);
        }
    }
}

int main()
{
    cin >> rows >> cols;
    vector<vector<int>> grid(rows, vector<int>(cols, 0));

    for (int r = 0; r < rows; r++) {
        for (int c = 0; c < cols; c++) {
            cin >> grid[r][c];
        }
    }

    for (int r = 0; r < rows; r++) {
        for (int c = 0; c < cols; c++) {
            if (grid[r][c] == 5) {   // 预打标  grid[r][c] = 0, 表示 (r, c) 为边界坐标
                for (int dr = -1; dr < 2; dr++) {
                    for (int dc = -1; dc < 2; dc++) {
                        int nr = r + dr, nc = c + dc;
                        if (valid(nr, nc) && grid[nr][nc] == 1) {
                            grid[nr][nc] = 0;   // 标记为可能是边界
                        }
                    }
                }
            }
        }
    }

    int result = 0;
    for (int r = 0; r < rows; r++) {
        for (int c = 0; c < cols; c++) {
            if (grid[r][c] == 0) {
                dfs(grid, r, c);
                result++;
            }
        }
    }

    cout << result << endl;
}    

希望这个专栏能让您熟练掌握算法, 🎁🎁🎁 立即订阅

整理题解不易, 如果有帮助到您,请给点个赞 ‍❤️‍ 和收藏 ⭐,让更多的人看到。🙏🙏🙏

#面经##春招##校招##华为##C++#
C++笔试真题题解 文章被收录于专栏

笔试真题题解

全部评论

相关推荐

评论
3
1
分享

创作者周榜

更多
牛客网
牛客企业服务