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

运行结果如下：