剑指offer-顺时针打印矩阵

263 阅读1分钟

顺时针打印矩阵leetcode54

给定一个包含 m x n 个元素的矩阵(m 行, n 列),按顺时针螺旋顺序,返回矩阵中的所有元素。

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

思路

class Solution(object):
    def spiralOrder(self, matrix):
        res = []
        while matrix:
            res += matrix.pop(0)  # pop移除第一行,加入res
            matrix = list(map(list, zip(*matrix)))[::-1]
        return res
"""
:type matrix: List[List[int]]
:rtype: List[int]
矩阵逆时针旋转:先转置,再上下逆置。
因此,一个想法是:每次将矩阵的第一行取出来,放入res,对剩下的矩阵逆时针旋转90°(转置再上下逆置),重复以上步骤(每次拿的都是第一行)。

用zip(*matrix)转置,zip(*iterables)返回一个迭代器,
						该迭代器枚举出一些tuple,其元素为各参数相应位置的那组元素。
用map(list, zip(*matrix))输出转置结果的多个小列表,map(function, iterable, ...)
用list(map(list, zip(*matrix)))组成一个大列表(矩阵)
用[::-1]逆转

将第一行加入res,并从矩阵中删除。此时res=[1,2,3],Matrix如下:
[
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
转置:
[
 [ 4, 7 ], 
 [ 5, 8 ],
 [ 6, 9 ]
]
上下逆置:
[
 [ 6, 9 ], 
 [ 5, 8 ],
 [ 4, 7 ]
]
"""