原书上写的解太麻烦了,而且还没有扩展性(比如从右上角开始打印,或者逆时针打印)。
此外多个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) 回帖