“之” 字形打印矩阵-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)
运行结果如下: