需求
给定一个 n × n 的二维矩阵表示一个图像。
将图像顺时针旋转 90 度。
说明
你必须在原地旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要使用另一个矩阵来旋转图像。
示例1
给定 matrix =
[
[1,2,3],
[4,5,6],
[7,8,9]
],
原地旋转输入矩阵,使其变为:
[
[7,4,1],
[8,5,2],
[9,6,3]
]
示例2
给定 matrix =
[
[ 5, 1, 9,11],
[ 2, 4, 8,10],
[13, 3, 6, 7],
[15,14,12,16]
],
原地旋转输入矩阵,使其变为:
[
[15,13, 2, 5],
[14, 3, 4, 1],
[12, 6, 8, 9],
[16, 7,10,11]
]
废话少说直接上代码:
方法一:
class Solution:
def rotate(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
length = len(matrix)
for i in range(length):
for j in range(i+1,length):
temp = matrix[i][j]
matrix[i][j] = matrix[j][i]
matrix[j][i] = temp
for i in range(lenth):
matrix[i] = matrix[i][::-1]
方法二:
class Solution:
def rotate(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
matrix[:] = list(map(list, zip(*matrix[::-1])))
总结
第一种方法,使用的是归纳总结的方法,要想实现90°旋转,核心思想是先将n x n的矩阵进行转置操作,第一个for循环就是做的转置矩阵的操作,然后将转置后的矩阵里面的每个列表进行一个反转,第二个for循环就是做的矩阵内列表的反转,这就完成了矩阵内元素的90°旋转。
第二种方法,先将矩阵内的列表进行反转,注意是反转列表,列表里的元素相对位置并没有改变,然后对矩阵进行拆包获得n个列表,然后对这n个列表进行zip操作,取每个列表的同一个索引位置的元素重新组成新的元组,这个时候元组还是n个,然后使用map将所有元组转换成列表,这个时候其实经过map处理后外层还不是列表,而是一个迭代器(Python3中的函数式编程的特性),所以我们最后还要将它转化成列表的形式,也就是我们最后得出的矩阵。
至于运行时间,第二种方法更胜一筹,写法也比第一种简单。
但是如果想深刻理解算法,我推荐将第一种方法搞透即可。