首页 > 剑指offer 29 顺时针打印矩阵(可逆时针,任一点开始)
头像
谁能给我个offer啊
编辑于 2021-08-09 19:13
+ 关注

剑指offer 29 顺时针打印矩阵(可逆时针,任一点开始)

原书上写的解太麻烦了,而且还没有扩展性(比如从右上角开始打印,或者逆时针打印)。

此外多个for循环的起止点也让人头晕,每个方向都要单独判断步进方向,因此代码也缺少优雅性。

这里我用directions控制x,y的步进方向,一目了然。

# data 为输入矩阵 这里我就自定义随便写了 有需要可以自己调整
data = [[1, 2, 3, 4, 5],
        [6, 7, 8, 9, 10],
        [11, 12, 13, 14, 15]]

# m 为行数,与x 相关    n为列数,与y相关
m = len(data)
n = len(data[0])

# data的维度 以及x y的移动方向要搞清楚 然后就明了了
# 设定一个 变换方向 [0, 1] 就代表 x方向不移动 y方向➕1
# 可以根据需要 手动调整成逆时针
directions = [[0, 1], [1, 0], [0, -1], [-1, 0]]  # → ↓ ← ↑

#建立一个visited矩阵 如果打印过了 就扭转方向
visited = [[False] * n for i in range(m)]

# 由于本题是要求从左上角开始 顺时针打印 那么 起始点就是 x=y=0  方向就是 [0, 1]
direc_id = 0
one_direction = directions[direc_id]
x = 0
y = 0

#记录要打印数字的总量 后面while用得到 如果都打印满了 就结束打印过程
cnt = m*n

while cnt:
    visited[x][y] = True
    print(data[x][y], ', ', end='')
    cnt -= 1

    tmp_x = x + one_direction[0] 
    tmp_y = y + one_direction[1]
    if 0 <= tmp_x < m and 0 <= tmp_y < n and visited[tmp_x][tmp_y] is not True:
        #不越界而且 还没打印过 就可以把坐标移动到下一个点上
        x = tmp_x
        y = tmp_y
    else:
        # 撞墙了 调整方向
        direc_id = (direc_id + 1) % 4
        one_direction = directions[direc_id]
        x += one_direction[0]
        y += one_direction[1]

最后输出1 , 2 , 3 , 4 , 5 , 10 , 15 , 14 , 13 , 12 , 11 , 6 , 7 , 8 , 9 , 

全部评论

(0) 回帖
加载中...
话题 回帖

相关热帖

近期精华帖

热门推荐