【题目】
给定一个矩阵 matrix,按照“之”字形的方式打印这个矩阵,例如:
1 2 3 4
4 5 6 8
9 10 11 12
“之”字形打印的结果为:1, 2, 5, 9, 6, 3, 4, 7, 10, 11, 8, 12
要求额外空间复杂度为O(1)
【基本思路】
- 上坐标(tR1,tC1)初始化为(0,0),先沿着矩阵第一行移动(tC1++),当到达第一行最右边的元素后,再沿着矩阵最后一列移动(tR1++).
- 下坐标(tR2,tC2)初始化为(0,0),先沿着矩阵第一列移动(tR2++),当到达第一列最下边的元素后,再沿着矩阵最后一行移动(tC2++).
- 上坐标和下坐标同时移动,每次移动后,上坐标和下坐标的连线就是矩阵的一条斜线,打印斜线上的元素即可.
- 用一个boolean型变量flag控制打印的方向.
下面是使用python3.5实现的代码。
def printMatrixZigZag(m):
def printLevel(m, tR1, tC1, tR2, tC2, flag):
if flag:
while tR2 != tR1 - 1:
print(m[tR2][tC2], end=' ')
tR2 -= 1
tC2 += 1
else:
while tR1 != tR2 + 1:
print(m[tR1][tC1], end=' ')
tR1 += 1
tC1 -= 1
tR1 = tC1 = tR2 = tC2 = 0
dR = len(m) - 1
dC = len(m[0]) - 1
flag = True
while tR1 <= dR and tC2 <= dC:
printLevel(m, tR1, tC1, tR2, tC2, flag)
tR1 = tR1 if tC1 != dC else tR1 + 1
tC1 = tC1 + 1 if tC1 != dC else tC1
tC2 = tC2 if tR2 != dR else tC2 + 1
tR2 = tR2 + 1 if tR2 != dR else tR2
flag = not flag