Mrfei 发表于 2015-12-2 15:19:52

leetcode-Spiral Matrix II 螺旋矩阵2之python大法好,四行就搞定,你敢信?

   Spiral Matrix II
  
  螺旋矩阵

  Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
  For example,
Given n = 3,
  
You should return the following matrix:

[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
  
  看博客园有人面试碰到过这个问题,长篇大论看得我头晕。跑去leetcode讨论区,看来看去还是我python大法牛逼,四行就搞定了你敢信。。

  | 1 | 2 | 3 |
  | 8 | 9 | 4 |
  | 7   6 | 5 |
  如果看不明白的话,大致过程像上面这样,颜色相同表示在同一步骤里面执行
  



>>> def generateMatrix(n):
...   A, low = [], n*n+1
...   while low > 1:
...         low, high = low - len(A), low
...         A = + list(zip(*A[::-1]))# python3.3 语法 与2.7会有差别
...         print(A)
...   return A
...
>>> generateMatrix(1)
[[]]
[]
[]
>>> generateMatrix(2)
[[]]
[]
[,
(4,)]
[,
(4, 3)]
[, (4, 3)]
>>> generateMatrix(3)
[[]]
[]
[,
(9,)]
[,
(9, 8)]
[,
(9, 6),
(8, 7)]
[,
(8, 9, 4),
(7, 6, 5)]
[, (8, 9, 4), (7, 6, 5)]

  
页: [1]
查看完整版本: leetcode-Spiral Matrix II 螺旋矩阵2之python大法好,四行就搞定,你敢信?