LeetCode使用Python实现旋转图像

需求

给定一个 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中的函数式编程的特性),所以我们最后还要将它转化成列表的形式,也就是我们最后得出的矩阵。

至于运行时间,第二种方法更胜一筹,写法也比第一种简单。

但是如果想深刻理解算法,我推荐将第一种方法搞透即可。

猜你喜欢

转载自blog.csdn.net/june_young_fan/article/details/84842513