题解 | #顺时针打印矩阵#

顺时针打印矩阵

http://www.nowcoder.com/practice/9b4c81a02cd34f76be2659fa0d54342a

借助一个辅助矩阵lab来记录元素是否被遍历,或者是否到了边界,然后按照题目规则去遍历整个矩阵,并保存到result中。

class Solution:
    def printMatrix(self, matrix: List[List[int]]) -> List[int]:
        # 判断matrix不为空
        if len(matrix[0]) != 0:
            # 构建matrix中每个元素的标签矩阵lab,表示该元素是否被遍历,初始值为True
            # lab比matrix首尾多出两行两列,并初始化为False,便于判断坐标移动
            lab = [[False for _ in range(len(matrix[0]) + 1)]]
            for i in matrix:
                x = [False]
                for j in matrix[0]:
                    x.append(True)
                x.append(False)
                lab.append(x)
            lab.append([False for _ in range(len(matrix[0]) + 1)])
            # 构建输出用的list,命名为result
            result = []
            # towhere表示坐标换行移动方向,0为向左向右向下移动,1为向上移动,初始值为0。
            towhere = 0
            # 初始化坐标位置,x为行标,y为列标
            x = 1
            y = 1
            # 开始遍历整个matrix
            for i in range(len(matrix) * len(matrix[0])):
                # 将当前坐标位置的数保存到reslut
                result.append(matrix[x - 1][y - 1])
                # 保存完了将lab中对应位置标签改为False,表示已遍历
                lab[x][y] = False
                # 开始移动坐标,先判断方向
                if towhere == 0:
                    # 当前是向右下走,先考虑能不能向右走,若向右走,该位置没有被遍历则y+1
                    if lab[x][y + 1]:
                        y += 1
                    # 不能向右走,考虑向下走
                    else:
                        # 判断向下走,该位置有没有被遍历,没被遍历x+1
                        if lab[x + 1][y]:
                            x += 1
                        # 不能向下也不能向右走,考虑往左走
                        else:
                            # 判断向左走,该位置没有被遍历,则y-1
                            if lab[x][y - 1]:
                                y -= 1
                            # 不能向左走,则向上走
                            else:
                                # 判断向上走,该位置没有被遍历x-1
                                if lab[x - 1][y]:
                                    x -= 1
                                    # 此时将towhere改为1,因为此时改向上走了
                                    towhere = 1
                # 若此时向上走
                else:
                    # 判断向上走,该位置没有被遍历x-1
                    if lab[x - 1][y]:
                        x -= 1
                    # 否则向右走,并将towhere改为0,因为接下去走到头就要向下换行
                    else:
                        y += 1
                        towhere = 0
            return result
        # 当matrix为空,直接返回空list
        else:
            return []
                
            
                
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务