码迷,mamicode.com
首页 > 其他好文 > 详细

刷题 11/05

时间:2019-11-07 09:29:02      阅读:98      评论:0      收藏:0      [点我收藏+]

标签:长度   ever   等于   交换   return   刷题   matrix   算法   python   

从昨天开始刷题,但是昨天比较懒,没有总结一下。今天开始总结一下刷题什么的遇到的bug和之前不注意的事情。

顺时针旋转矩阵

有一个NxN整数矩阵,请编写一个算法,将矩阵顺时针旋转90度。
给定一个NxN的矩阵,和矩阵的阶数N,请返回旋转后的NxN矩阵,保证N小于等于300。

很简单的矩阵变换,我是通过两次矩阵变换得到的,首先对角线交换,然后左右翻转就可以了。

class Rotate:
    def rotateMatrix(self, mat, n):
        c = [[0 for i in range(0,n)] for j in range(0,n)]
        for i in range(0,n):
            for j in range(0,n):
                c[i][j] = mat[j][i]
        for i in range(0,n):
            c[i].reverse()
        return(c)

但是实际上之前遇到了一个问题,就是只用python自带的库并不知道怎么生成矩阵。
一开始是用下面的代码生成的
c = mat
后来意识到这样只是共享内存,然后就用这样
matrix = [[0]*n]*n
然后发现这样只是重复一行的内容,重复n行,更加完蛋。
最后
c = [[0 for i in range(0,n)] for j in range(0,n)]
据说是因为python浅引用的问题。

后来看到还有一个解法,就是python的zip函数。
zip函数是打包函数,返回对象是zip,实际上是将列表中元素打包成元组。我们可以用list转化为列表。

a = [1,3,4]
b = [4,5,6]
c = [4,5,6,7,8]
zip(a,b)
print(list(zip(a,b)))
print(list(zip(a,c)))

返回值长度为原列表中较短列表的长度。

[(1, 4), (3, 5), (4, 6)]
[(1, 4), (3, 5), (4, 6)]

zip(*)函数为解压函数,同样返回zip内容,需要用list转化为列表。

a = [1,3,4]
b = [4,5,6]
zipped = zip(a,b)
a1,a2 = zip(*zipped)
print(list(a1))
print(list(a2))

返回值为原来的列表。

[1, 3, 4]
[4, 5, 6]

然后就是本题目
list(zip(*mat))
将方阵mat转化为一个zip对象,再将其转化为列表。例如

z = [[1,2,3],[4,5,6],[7,8,9]]
a = list(zip(*z))
print(a)

输出
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
然后列表中[::-1]表示逆序,所以直接将a的每一列元组先转化为列表,再对列表进行逆序操作就可以得到。

z = [[1,2,3],[4,5,6],[7,8,9]]
for i in zip(*z):
    print(list(i[::-1]))

返回结果

[7, 4, 1]
[8, 5, 2]
[9, 6, 3]

回到原题,可以用直接生成列表

class Rotate:
    def rotateMatrix(self, mat, n):
        a = list(zip(*mat))
        for i in range(0,len(a)):
            a[i] = list(a[i])
            a[i].reverse()
        return(a)

看到有一个快速解法,即
return([x[::-1] for x in zip(*mat)])
确实这个也可以通过,但是我自己在jupyter的python3 和 在线某网站的python2中跑下来的中间返回的都是元组,所以不知道系统是怎么判定是否通过的。很迷。

刷题 11/05

标签:长度   ever   等于   交换   return   刷题   matrix   算法   python   

原文地址:https://www.cnblogs.com/Komnenos/p/python_zip.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!