PAT (Basic)1027 打印沙漏 （Python实现）

打印沙漏

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

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

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

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

输入格式：

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

输出格式：

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

输入样例

19 *

输出样例：

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

---

分析：

观察每个沙漏，看每个沙漏由多少个符号组成。发现它们的递增规律是1、7、17、...、2*k^2-1,那么找出这个K就能把沙漏打印出来了。

---

代码

import math

N, sign = input().split(" ")
N = int(N)
k = int(math.sqrt((N+1)/2))
# 求出剩下的符号数
surplus = N-2*k**2+1

# 打印沙漏
for i in range(k):
    print(" "*i, end="")
    print(sign*(2*(k-i)-1))
for i in range(2,k+1):
    print(" "*(k-i), end="")
    print(sign*(2*i-1))
print(surplus)