“之” 字形打印矩阵-python3

最新推荐文章于 2022-12-17 22:36:39 发布

原创

最新推荐文章于 2022-12-17 22:36:39 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍如何使用Python3实现'之'字形打印矩阵。通过定义一个布尔变量来控制打印方向，实现了上下来回交替的打印效果。

该问题要先打印指定的一条斜线。

由于之字形打印是上下来回交替着打印，所以要定义一个bool变量来指示打印的方向。

代码如下：

#coding=utf-8
'''
“之” 字形打印矩阵
【 题目】 给定一个矩阵matrix， 按照“之” 字形的方式打印这
个矩阵， 例如： 1 2 3 4
               5 6 7 8
              9 10 11 12
“之” 字形打印的结果为： 1， 2， 5， 9， 6， 3， 4， 7， 10， 11，
8， 12
【 要求】 额外空间复杂度为O(1)。
'''
import numpy as np
def printMatrixZigZag(matrix):
    aR,aC=0,0
    bR,bC=0,0
    endR=matrix.shape[0]-1
    endC=matrix.shape[1]-1
    fromUp=False
    while aR!=endR+1:
        printLevel(matrix,aR,aC,bR,bC,fromUp)
        # 下面四句改变顺序会报错(if判断条件的原因)
        aR=aR+1 if aC==endC else aR
        aC=aC if aC==endC else aC+1
        bC = bC + 1 if bR == endR else bC
        bR=bR if bR==endR else bR+1

        fromUp=not fromUp
def printLevel(m,aR,aC,bR,bC,fromUp):
    if fromUp:
        while aR!=bR+1: #边界
            p