“之”字形打印矩阵
【題目】 給定一個(gè)矩陣matrix,按照“之”字形的方式打印這 個(gè)矩陣,例如:1 2 3 45 6 7 89 10 11 12 “之”字形打印的結(jié)果為:1,2,5,9,6,3,4,7,10,11, 8,12
【要求】 額外空間復(fù)雜度為O(1)
思路:
1、上坐標(biāo)(tR,tC) 初始值為(0,0),先沿著第一行移動(dòng)(tC?+=1),達(dá)到第一行最右邊的元素后,再沿著矩陣最后一列移動(dòng)(tR+=1)
2、下坐標(biāo)(dR,dC)初始值為(0,0), 先沿著第一列移動(dòng)(dR +=1), 達(dá)到第一列最下邊的元素時(shí),再沿著矩陣最后一行移動(dòng)(dC+=1)
3 、上坐標(biāo)和下坐標(biāo)同步移動(dòng),每次移動(dòng)后的上坐標(biāo)和下坐標(biāo)的連線(xiàn)就是矩陣中的一條斜線(xiàn),打印斜線(xiàn)上的元素即可
4、如果上次是從左向右上打印的,這次一定是從右上向左下打印,反之亦然,總之可以把打印的方向用boolearn值表示,每次取反
即可
def printMatrixZigZag(matrix):tR = 0tC = 0dR = 0dC = 0endR = len(matrix) - 1endC = len(matrix[0]) - 1fromUp = Falsewhile tR!=endR+1:printlevel(matrix,tR,tC,dR,dC,fromUp)if tC != endC:tC = tC + 1else:tC = tCif tC != endC:tR = tRelse:tR = tR + 1if dR != endR:dR = dR + 1else:dR = dRif dR != endR:dC = dCelse:dC = dC + 1fromUp = bool(1-fromUp)def printlevel(matrix,tR,tC,dR,dC,fromUp):if fromUp:while tR != dR + 1:print(matrix[tR][tC])tR +=1tC -=1else:while dR != tR - 1:print(matrix[dR][dC])dR -=1dC +=1matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]] printMatrixZigZag(matrix)?
總結(jié)