“之” 字形打印矩阵-python3

《“之” 字形打印矩阵-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: #边界
            print(m[aR,aC],end=" ")
            aR+=1
            aC-=1
        # print("")
    else:
        while bR!=aR-1: #边界
            print(m[bR,bC],end=" ")
            bR-=1
            bC+=1
        # print("")
if __name__=="__main__":
    matrix=np.arange(1,17).reshape(4,4)
    print(matrix)
    print("="*25)
    printMatrixZigZag(matrix)

运行结果如下:

《“之” 字形打印矩阵-python3》

    原文作者:Z字形编排问题
    原文地址: https://blog.csdn.net/xujiao1179/article/details/85640021
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞