PTA L1-002 打印沙漏(Python实现) 测试点全过

题目

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

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

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

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

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

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

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

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

题解

# 输入个数和*
n, c = input().split()
n = int(n)

# 判断最大沙漏
i = 1;
while True:
    if 2 * i**2 - 1 > n:
        i -= 1;
        break
    i += 1
    
# 打印上部分
for j in range(i, 0, -1):
    print(' ' * (i - j), end = '')
    print(c * (2 * j - 1))

# 打印下部分
for j in range(2, i + 1):
    print(' ' * (i - j), end = '')
    print(c * (2 * j - 1))

# 打印没用掉的符号数
print(n - (2 * i**2 - 1))