1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
| class Solution(object):
def generateMatrix(self, n):
"""
:type n: int
:rtype: List[List[int]]
"""
if n == 0:
return []
rs = [[0] * n for i in xrange(n)]
left = up = 0
right = down = n - 1
x = 1
while right >= left and down >= up:
if right >= left and down >= up:
for i in xrange(left, right + 1):
rs[up][i] = x
x += 1
up += 1
if down >= up and right >= left:
for i in xrange(up, down + 1):
rs[i][right] = x
x += 1
right -= 1
if right >= left and down >= up:
for i in xrange(right, left - 1, -1):
rs[down][i] = x
x += 1
down -= 1
if down >= up and right >= left:
for i in xrange(down, up - 1, -1):
rs[i][left] = x
x += 1
left += 1
return rs
|