PTA 团体程序设计天梯赛-练习集--L1-002 打印沙漏 (20分)（Python3)

题目

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

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

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

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

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

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

输入样例:
19 *
输出样例:

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

解答

import sys

InputInfo = input ("")  #输入指定格式
InputList = InputInfo.split(' ')  #按照空格进行切片
Cont = int(InputList[0])	#索引0为个数
sig = InputList[1] 		#索引1为符号
tem = ' '	#设置空格

if type(Cont) != type (1) or Cont<=0:	#判断格式是否正确
    print("Something goes wrong, you didn't input a positive integer")
    sys.exit()

#判断正确的行数
all_num = 0		
for i in range(1,Cont):
    all_num = 2*i*i-1	#根据等差数列前n项和化简
    if all_num > Cont:	#当前总和大于指定的数量
        i = i-1	#返回上一层
        break		#退出循环
acc = 2*i*i-1	#计算当前已经使用的字符个数
rest = Cont-acc	#剩下个数

max_n = 2*i-1		#当前层个数最多的
for x in range(1,i+1):		#每层开始循环
    tem_num = max_n-(2*(i-x+1)-1)	#空格数量，最大数量减去当前层数
    tem_all = tem_num*tem	#创建空格字符串
    ans_inf = tem_all[0:tem_num//2]+(2*(i-x+1)-1)*sig	#切片空格前半部分且凭借指定字符
    print(ans_inf)	#打印当前层
    
#同上
for x in range(2,i+1):
    tem_num = max_n-(2*x-1)
    tem_all = tem_num*tem
    ans_inf = tem_all[0:tem_num//2]+(2*x-1)*sig
    print(ans_inf)
print(rest,end = '')	#打印剩下数量

问题：

1.打印每一层时，只需打印左半部分，右半部分不需要，否则会报错。

2.返回值非0(为解决)，希望大家能提供帮助。