【Python】旋转打印各种矩形

打印旋转矩阵应该是很经典的算法问题了。题目描述如下:

给定一个m * n要素的矩阵。按照螺旋顺序,返回该矩阵的所有要素。

思路:1,先定义矩阵的左上和右下的坐标,然后通过两个坐标来打印这一圈矩阵;

   2,将左上的坐标下右下移动,右下的坐标向左上移动,来缩小打印圈,进行下一圈矩阵的打印;

   3,一直缩小打印直到结束。

代码:

def print_circle(matrix,up_hang,up_lie,down_hang,down_lie):
    result=[]
    
    if up_lie==down_hang and down_hang==down_lie:  # 若只有一个元素
        result.append(matrix[up_hang][up_lie])
    elif up_lie==down_hang or up_lie==down_lie:    #若只有一行或一列元素
        if up_lie==down_hang:
            while up_lie <= down_lie:
                result.append(matrix[up_hang][up_lie])
                up_lie+=1
        elif up_lie==down_lie:
            while up_hang <=down_hang:
                result.append(matrix[up_hang][up_lie])
                up_hang+=1
#  return result        #注意对齐方式,其决定了作用的区间范围,很关键        
        return result      
    i=up_hang
    j=up_lie
    while jup_lie:
        result.append(matrix[i][j])
        j-=1
    while i>up_hang:
        result.append(matrix[i][j])
        i-=1
    return result

#matrix=[[1,2,3],[4,5,6],[7,8,9],[10,11,12],[13,14,15],[16,17,18]]
#matrix=[[1,2,3,4,5],[6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20],[21,22,23,24,25]]
matrix=[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]]
#matrix=[[1,2,3],[4,5,6],[7,8,9]]
#matrix=[[1,2],[3,4]]
re_mat=[]
up_hang=0
up_lie=0
down_hang=3
down_lie=3
if down_hang>down_lie:
    flag=down_lie
else:
    flag=down_hang
while flag!=0:             #flag决定了一个矩阵需要打印多少圈
    temp=print_circle(matrix,up_hang,up_lie,down_hang,down_lie)
    re_mat.extend(temp)
    up_hang+=1
    up_lie+=1
    down_hang-=1
    down_lie-=1
    flag=flag/2
print(re_mat)
总结:python对于代码的对齐方式要求的比较严格,对齐方式直接决定了函数或者判断条件的作用域,要重视啊。

你可能感兴趣的:(算法,Python)