PTA 打印沙漏 python

PTA数据结构能力自测 自测-1 打印沙漏

本题要求你写个程序把给定的符号打印成沙漏的形状。例如给定17个“*”，要求按下列格式打印

*****
 ***
  *
 ***
*****

所谓“沙漏形状”，是指每行输出奇数个符号；各行符号中心对齐；相邻两行符号数差2；符号数先从大到小顺序递减到1，再从小到大顺序递增；首尾符号数相等。

给定任意N个符号，不一定能正好组成一个沙漏。要求打印出的沙漏能用掉尽可能多的符号。

输入格式:

输入在一行给出1个正整数N（≤1000）和一个符号，中间以空格分隔。

输出格式:

首先打印出由给定符号组成的最大的沙漏形状，最后在一行中输出剩下没用掉的符号数。

输入样例:

19 *

输出样例:

*****
 ***
  *
 ***
*****
2

解题思路：

1. 将输入的数值加1后除以2取整数，该做法的原因是将沙漏分成上下两部分，每部分镜像重叠，但实际上是缺少了沙漏的中间值数字1，所以先补上1后除以2取整，先完成沙漏的上半部分。
2. 利用等差数列求和的限定条件构建一个列表
3. 将列表里的数字打印出来即为最终结果

代码如下

n,m = input().split()
n = int(n)
target = (n+1)//2
list1 = []
sums = 0
for i in range(1,target+1,2):
    sums = sums + i
    if sums > target:#先判断，再添加列表，反过来的话会在number处出现负数
        break
    list1.append(i)
list2 = list1[1:]#因为下半部分没有1了，所以从列表的下标1开始取，跳过数字1
total = sum(list1) + sum(list2)
number = n - total
for j in range(len(list1)):
    print(' '*j+m*list1[int(len(list1)-j-1)])
for k in range(len(list2)):
    print(' '*int(len(list2)-k-1)+m*list2[k])
print(number)