PAT~乙级~1050 螺旋矩阵 ~python

绝地灬酷狼 2021-11-27 06:28 327阅读 0赞

题目描述：

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m≥n；且 m−n 取所有可能值中的最小值。

### 输入格式： ###

输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 104，相邻数字以空格分隔。

### 输出格式： ###

输出螺旋矩阵。每行 n 个数字，共 m 行。相邻数字以 1 个空格分隔，行末不得有多余空格。

### 输入样例： ###

12
    37 76 20 98 76 42 53 95 60 81 58 93

### 输出样例： ###

98 95 93
    42 37 81
    53 20 76
    58 60 76

\----------------------------------------------------------答题区----------------------------------------------------------

最主要的是把一步步的逻辑理清，代码里有一些注释。

import math
    num=int(input())
    max_flag=num
    m,n=0,0
    for i in range(math.floor(math.sqrt(num)),num+1):
        temp_div=num//i
        if(num%i==0 and i>=temp_div and (i-temp_div)<max_flag):
            m=i
            n=temp_div
            max_flag=m-n
    s=list(map(int,input().split()))
    s.sort(reverse=True)
    if(n==1):
        for i in s:
            print(i)
    else:
        res=[[0]*n for i in range(m)]
        hang=0
        lie=-1
        hang_flag=1
        you=1
        hang_weibiao=m-1
        lie_weibiao=n-1
        hang_qianbiao=1
        lie_qianbiao=0
        for i in s:
            if(hang_flag==1 and you==1):
                #行 向右
                lie+=1
                res[hang][lie]=i
                if(lie==lie_weibiao):
                    hang_flag=0
                    lie_weibiao-=1
            elif(hang_flag==0 and you==1):
                #列 向下
                hang+=1
                res[hang][lie]=i
                if(hang==hang_weibiao):
                    hang_flag=1
                    you=0
                    hang_weibiao-=1
            elif(hang_flag==1 and you==0):
                #行 向左
                lie-=1
                res[hang][lie]=i
                if(lie==lie_qianbiao):
                    you=0
                    hang_flag=0
                    lie_qianbiao+=1
            elif(hang_flag==0 and you==0):
                #列 向上
                hang-=1
                res[hang][lie]=i
                if(hang==hang_qianbiao):
                    hang_flag=1
                    hang_qianbiao+=1
                    you=1
        
        for i in range(len(res)):
            for j in range(len(res[i])):
                if(j==0):
                    print(res[i][j],end='')
                elif(j==len(res[i])-1):
                    print(" "+str(res[i][j]),end='\n')
                else:
                    print(" "+str(res[i][j]),end='')

维护了一个PAT测试点问题的共享文档，欢迎大家把遇到的问题写在上面。

一起写文档：[https://yiqixie.com/d/home/fcACe6xZ2aroPxFQrQZ05Cw32][https_yiqixie.com_d_home_fcACe6xZ2aroPxFQrQZ05Cw32]  
github链接：[https://github.com/Makerdd/PAT-yiji][https_github.com_Makerdd_PAT-yiji]

[https_yiqixie.com_d_home_fcACe6xZ2aroPxFQrQZ05Cw32]: https://yiqixie.com/d/home/fcACe6xZ2aroPxFQrQZ05Cw32
[https_github.com_Makerdd_PAT-yiji]: https://github.com/Makerdd/PAT-yiji