python基本語法3.10--綜合練習之旋轉矩陣

1.螺旋矩陣：給定一個m*n的矩陣。按照螺旋順序，返回該矩陣的所有要素。

[[1,2,3],

[4,5,6],

[7,8,9]]

返回：[1,2,3,6,9,8,7,4,5]

思路：從左往右，從上至下，從右至左，從下至上打印即可。

代碼：

def spiral_matrix(matrix, m, n):
    output = []
    startX= 0
    endX = n - 1
    startY = 0
    endY = m -1

    while True:
        if startX > endX:
            break
        for i in range(startX, endX + 1, 1):
            output.append(matrix[startY][i])
        startY += 1

        if startY > endY:
            break
        for j in range(startY, endY + 1, 1):
            output.append(matrix[j][endX])
        endX -= 1

        if startX > endX:
            break
        for i in range(endX, startX -1 , -1):
            output.append(matrix[endY][i])
        endY -= 1

        if startY > endY:
            break
        for j in range(endY, startY -1 , -1):
            output.append(matrix[j][startX])
        startX += 1
    return output

matrix = [[1,2,3],[4,5,6],[7,8,9],[10,11,12,13]]
for i in range(0,len(matrix)):
    print(matrix[i])

ans = spiral_matrix(matrix,len(matrix) ,len(matrix[0]))
print(ans)