LeetCode59-螺旋矩阵II

这两天一直在单曲循环周董的歌

才发现原来这么好听

词棒编曲也厉害

真的是要有才了

以前只是觉得他唱歌吐字不清楚

现在才发现写的歌这么有意境

粉了粉了

---

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%左右。