【LEETCODE】59-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 ]
]

题意：
给一个整数 n，生成一个方阵，由 1 到 n^2 的螺旋顺序构成

参考： http://www.cnblogs.com/zuoyuan/p/3769892.html
思路：
用count去在该有的位置填入该有的数字，当count满足n＊n时则返回最终结果
用 up right down left 分别去当四条边的指针，填满一条边后进行更新
direct 是除4的余数，用来控制目前应该填入哪条边了

class Solution(object):
def generateMatrix(self, n):
"""
:type n: int
:rtype: List[List[int]]
"""
if n==0:
return []

matrix=[[n for i in range(n)]for j in range(n)]

left=0
up=0
right=n-1
down=n-1

direct=0
count=0

while True:
if direct==0:                           #此时为 上边
for i in range(left,right+1):       #up控制上边 当前的行，left为当前起点，right为终点
count+=1
matrix[up][i]=count
up+=1                               #写完一行，要进行更新

if direct==1:                           #右边
for i in range(up,down+1):
count+=1
matrix[i][right]=count
right-=1

if direct==2:                           #下边
for i in range(right,left-1,-1):
count+=1
matrix[down][i]=count
down-=1

if direct==3:                           #左边
for i in range(down,up-1,-1):
count+=1
matrix[i][left]=count
left+=1

if count==n*n:                          #终止条件
return matrix

direct=(direct+1)%4                     #更新边的方向