天天看點

順時針列印矩陣

題目描述

輸入一個矩陣,按照從外向裡以順時針的順序依次列印出每一個數字,例如,如果輸入如下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.

解答

# coding:utf-8

class Solution:
    # matrix類型為二維清單,需要傳回清單
    def printMatrix(self, matrix):
        # write code here
        pro = matrix
        l1 = []
        while 1:
            # 将矩陣第一行加到l1裡面
            l1.extend(pro[0])
            # 如果矩陣大于一行,就取剩下的行
            if len(pro) > 1:
                pro = pro[1:]
            else:
                break
            # 90°旋轉矩陣
            # 雙層清單推導式,[[r[i] for r in pro]]裡面的i是for i in range(len(pro[0])-1, -1, -1)的值
            pro = [[r[i] for r in pro] for i in range(len(pro[0])-1, -1, -1)]
        return l1

matrix = [[1,2,3],[4,5,6],[7,8,9]]
print Solution().printMatrix(matrix)      

結束!