PTA 1036 跟奥巴马一起编程 (15分) (python 3)

美国总统奥巴马不仅呼吁所有人都学习编程，甚至以身作则编写代码，成为美国历史上首位编写计算机代码的总统。2014 年底，为庆祝“计算机科学教育周”正式启动，奥巴马编写了很简单的计算机代码：在屏幕上画一个正方形。现在你也跟他一起画吧！

输入格式：
输入在一行中给出正方形边长 N（3≤N≤20）和组成正方形边的某种字符 C，间隔一个空格。

输出格式：
输出由给定字符 C 画出的正方形。但是注意到行间距比列间距大，所以为了让结果看上去更像正方形，我们输出的行数实际上是列数的 50%（四舍五入取整）。

输入样例：

10 a

输出样例：

aaaaaaaaaa
a      	 a
a        a
a      	 a
aaaaaaaaaa

代码

t=input().split()
n=int(t[0])
c=t[1]
a=int(t[0])//2+int(t[0])%2
if 3<=n<=20:
    for i in range(a):
        if i==0 or i==a-1:
            print(c*n)
        else:
            print(c,end='')
            print(c.rjust(n-1,' '))