常见面试算法题-图像物体的边界

■ 题目描述

【图像物体的边界】

给定一个二维数组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

说明

参考题目描述部分

以下代码为本人原创，可以供大家参考，若有不足之处，感谢指出！！！！

def put(i, j, a, pre, goal, boolean):
    for x, y in directions:
        newi = i + x
        newj = j + y
        if 0 <= newi < m and 0 <= newj < n and a[newi][newj] == pre:
            a[newi][newj] = goal
            if boolean:
                put(newi, newj, a, pre, goal, boolean)


m, n = list(map(int, input().split()))
matrix = [[] for _ in range(m)]
for i in range(m):
    matrix[i] = list(map(int, input().split()))
directions = [(0, 1), (0, -1), (1, 0), (-1, 0), (1, 1), (-1, -1), (-1, 1), (1, -1)]
for i in range(m):
    for j in range(n):
        if matrix[i][j] == 5:
            put(i, j, matrix, 1, 2, False)
ans = 0
for i in range(m):
    for j in range(n):
        if matrix[i][j] == 2:
            put(i, j, matrix, 2, 3, True)
            ans += 1
print(ans)