“之” 字形打印矩阵-python3 - Go语言中文社区

“之” 字形打印矩阵-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)

运行结果如下:

版权声明:本文来源CSDN,感谢博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/xujiao1179/article/details/85640021
站方申明:本站部分内容来自社区用户分享,若涉及侵权,请联系站方删除。
  • 发表于 2020-04-18 22:15:21
  • 阅读 ( 754 )
  • 分类:

0 条评论

请先 登录 后评论

官方社群

GO教程

猜你喜欢