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

才发现原来这么好听

词棒编曲也厉害

真的是要有才了

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

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

粉了粉了


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

LeetCode59-螺旋矩阵II

相关文章:

  • 2021-11-05
  • 2021-04-21
  • 2021-12-28
  • 2021-05-30
  • 2022-01-14
  • 2022-12-23
  • 2021-08-27
  • 2021-08-11
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2021-09-22
相关资源
相似解决方案