这两天一直在单曲循环周董的歌
才发现原来这么好听
词棒编曲也厉害
真的是要有才了
以前只是觉得他唱歌吐字不清楚
现在才发现写的歌这么有意境
粉了粉了
59-螺旋矩阵II
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
思路:
这一题是第54题螺旋矩阵的进阶版,但我觉得这一题其实和第54题的难度差不多,甚至更简单。对第54题螺旋矩阵还不熟悉的读者可以先看看我写的这篇文章。
程小新同学:LeetCode54-螺旋矩阵zhuanlan.zhihu.com
这一题我觉得可以理解为:将第54题的要求给变了一下。第54题的要求是:给定一个包含mxn个元素的矩阵(m行,n列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。要求变更为:请按照顺时针螺旋顺序,将给定矩阵的元素改为相应遍历的步数。所以本题只需在第54题上做少量修改即可。废话不多说,直接看代码:
class Solution(object):
def generateMatrix(self, n):
"""
:type n: int
:rtype: List[List[int]]
"""
# 首先初始化一个n*n的全0矩阵
initial_matrix = []
for index in range(n):
initial_matrix.append([0]*n)
# 初始化一个从1开始共计n*n的列表
initial_list = list(range(1, n*n+1))
# 定义点移动的方向集合,先后顺序代表优先级,题目要求顺时针旋转
move_direction = [[0, 1], [1, 0], [0, -1], [-1, 0]]
# 定义移动点的初始位置
start_pos_left, start_pos_right = [0, 0]
# 定义移动的步数
move_step = 0
while True:
# 移动的步数move_step刚好等于n*n,说明数据已经安放完毕,可直接退出返回答案了
if move_step >= n*n:
break
initial_matrix[start_pos_left][start_pos_right] = initial_list[move_step]
# 获取当前位置start_pos的可移动方向
next_direction = self.get_direction([start_pos_left, start_pos_right], move_direction, initial_matrix)
# 如果当前位置不能移动了,说明原matrix矩阵已经遍历完成;反之则可按给定方向移动
if next_direction is not None:
initial_matrix[start_pos_left][start_pos_right] = initial_list[move_step]
move_step += 1
start_pos_left += next_direction[0]
start_pos_right += next_direction[1]
while True:
if start_pos_left in range(n) and start_pos_right in range(n) and \
initial_matrix[start_pos_left][start_pos_right] == 0:
initial_matrix[start_pos_left][start_pos_right] = initial_list[move_step]
# 获取当前位置start_pos按求得next_direction方向移动的下一个坐标位置
move_step += 1
start_pos_left += next_direction[0]
start_pos_right += next_direction[1]
# 如果当前位置不符合要求,则要回到上一步位置,准备切换方向继续遍历
else:
start_pos_left -= next_direction[0]
start_pos_right -= next_direction[1]
move_step -= 1
break
else:
break
return initial_matrix
def get_direction(self, start_pos, move_direction, initial_matrix):
# 按照给定的4个方向遍历,按照方向的先后顺序如果有哪个方向可通过,则立即返回该方向;反之,则返回None,说明没有方向可通过了
for direction in move_direction:
left, right = start_pos
left += direction[0]
right += direction[1]
row = len(initial_matrix)
if left in range(row) and right in range(row) and initial_matrix[left][right] == 0:
return direction
return None
if __name__ == "__main__":
n = 3
final_matrix = Solution().generateMatrix(n)
print(final_matrix)
执行效率属于中上等吧,在70%左右。