L1-002打印沙漏问题——团体程序设计天梯赛-练习题

原题目：

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

思路分析：

首先将沙漏分成上下两部分考虑，将上半部分作为一个完整沙漏考虑，可以看出上半部分每行符号数量依次为1,3,5,7，……

假设沙漏的上半部分共有n行，这样上半部分的整个的符号总和为1+3+5+7+……+2n-1

作为一个等差数列考虑计算，则上半部分的符号和为n^2，因此整个沙漏的符号数为2n^2-1

从而根据输入的符号总个数N求得n

之后采用循环输出，先输出上半部分沙漏，再输出下半部分沙漏

即可得到正确的输出结果

完整代码：

import math
N, s = input().split()
N = int(N)
# 计算上半部分的总行数
n = math.floor(math.sqrt((N+1)/2))
# 打印上半部分
for i in range(n):
    spaces = i      # 左侧空格数
    s_numbers = 2*n-1-2*i       # 每行符号数
    print(" "*spaces+s*s_numbers)
# 打印下半部分
for i in range(2, n+1):
    spaces = n-i
    s_numbers = 2*i-1
    print(" "*spaces+s*s_numbers)
# 输出剩余符号数
c = N - (2*n**2-1)
print(c)