LeetCode(867)

题目描述

给定一个矩阵 A, 返回 A 的转置矩阵。

矩阵的转置是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。

示例 1:

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

示例 2:

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

提示:

  1. 1 <= A.length <= 1000
  2. 1 <= A[0].length <= 1000

解决方案

1.常规复制

class Solution:
    def transpose(self, A):
        """
        此方法的本质就是ans[c][r] = A[r][c]
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
        # R——A的长度,C——数组的长度
        R,C =len(A),len(A[0])
        ans = [[None]*R for _ in range(C)]
        # ans  [[None, None, None], [None, None, None], [None, None, None]]
        for r,row in enumerate(A):
            for c,val in enumerate(row):
                ans[c][r]= val
        return ans

def main():
    s  =Solution()
    b = s.transpose([[1,2,3],[4,5,6],[7,8,9]])
    print(b)
    #[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

if __name__ == "__main__":
    main()

2.zip函数一行搞定

class Solution:
    def transpose(self, A):
        """
        :type A: List[List[int]]
        :rtype: List[List[int]]
        """
       
        return list(zip(*A))

def main():
    s  =Solution()
    b = s.transpose([[1,2,3],[4,5,6],[7,8,9]])
    print(b)
    #[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

if __name__ == "__main__":
    main()

补充:zip函数的用法

描述

zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的对象,这样做的好处是节约了不少的内存。

我们可以使用 list() 转换来输出列表。

如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同,利用 * 号操作符,可以将元组解压为列表。

实例
>>>a = [1,2,3]
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b)     # 返回一个对象
>>> zipped
<zip object at 0x103abc288>
>>> list(zipped)  # list() 转换为列表
[(1, 4), (2, 5), (3, 6)]
>>> list(zip(a,c))              # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
 
>>> a1, a2 = zip(*zip(a,b))          # 与 zip 相反,zip(*) 可理解为解压,返回二维矩阵式
>>> list(a1)
[1, 2, 3]
>>> list(a2)
[4, 5, 6]
>>>
posted @ 2019-01-12 10:15  YanceDev  阅读(234)  评论(0编辑  收藏  举报