顺时针打印矩阵
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下4 X 4矩阵: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.
思路
- 迭代法
【黄申:迭代法,简单来说,其实就是不断地用旧的变量值,递推计算新的变量值】
对于本题中,任意时刻的矩阵的规模可以由四个顶点(startR,0),(startR,endC),(endR,0)以及(endR,endC)唯一确定,每次打印完一圈之后,startR,startC,endR,endC会被新的值替换(加1或减1),直至startR>endR 或者 startC > endC, 迭代终止。
本题中需要注意列矩阵和行距阵的判断,这里的列矩阵或是行距阵,不仅包括matrix本身,也包括在迭代过程中,matrix缩小后的矩阵规模。这类矩阵的规模只需要一组行或列的起始值和终止值来确定即可。
代码
# -*- coding:utf-8 -*-
class Solution:
# matrix类型为二维列表,需要返回列表
def printMatrix(self, matrix):
# write code here
if len(matrix) == 1:
return matrix[0]
startR,startC,endR,endC = 0,0,len(matrix)-1,len(matrix[0])-1
resList = []
while startR <= endR and startC <= endC:
self.printMatrixEdge(matrix,startR,startC,endR,endC,resList)
startR += 1
startC += 1
endR -= 1
endC -= 1
return resList
def printMatrixEdge(self,matrix,startR,startC,endR,endC,resList):
if startR == endR:
while startC <= endC:
resList.append(matrix[startR][startC])
startC += 1
elif startC == endC:
while startR <= endR:
resList.append(matrix[startR][startC])
startR += 1
else:
dR,dC = startR,startC
while dC < endC:
resList.append(matrix[dR][dC])
dR = dR
dC = dC + 1
while dR < endR:
resList.append(matrix[dR][dC])
dR = dR + 1
dC = dC
while dC > startC:
resList.append(matrix[dR][dC])
dR = dR
dC = dC - 1
while dR > startR:
resList.append(matrix[dR][dC])
dR = dR - 1
dC = dC
if __name__ == "__main__":
matrix = [[1],[2],[3],[4],[5]]
print Solution().printMatrix(matrix)