[Leetcode]54. Spiral Matrix @python

题目

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
You should return [1,2,3,6,9,8,7,4,5].

题目要求

给定一个矩阵,按照螺旋的方式读取矩阵中的元素,由内而外。

解题思路

此题借鉴南郭子綦的代码。这是一个使用状态转移的例子。
程序逻辑分为4个状态
1. 0表示向右边读取
2. 1表示向下读取
3. 2表示向左读取
4. 3表示向上读取
仅仅用状态还不够,因为要从最外圈向内读取,所以还需要四个指针:
1. up表示外圈的上行位置
2. down表示外圈的下行位置
3. left表示外圈的左列位置
4. right表示外圈的右列位置

代码

class Solution(object):
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        if len(matrix) == 0:
            return []
        ans = []
        direct = 0 # 0 for right, 1 for down, 2 for left, 3 for up
        up = left = 0
        down,right = len(matrix) - 1,len(matrix[0]) - 1
        while up <= down and left <= right:
            if direct == 0:
                for i in range(left,right + 1):
                    ans.append(matrix[up][i])
                up += 1
            elif direct == 1:
                for i in range(up,down + 1):
                    ans.append(matrix[i][right])
                right -= 1
            elif direct == 2:
                for i in range(left,right + 1)[::-1]:
                    ans.append(matrix[down][i])
                down -= 1
            else:
                for i in range(up,down + 1)[::-1]:
                    ans.append(matrix[i][left])
                left += 1
            direct = (direct + 1) % 4

        return ans
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值