PAT 1036 跟奥巴马一起编程 python

1036 跟奥巴马一起编程(15)（15 分）

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

输入格式：

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

输出格式：

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

输入样例：

10 a

输出样例：

aaaaaaaaaa
a        a
a        a
a        a
aaaaaaaaaa

作者: CHEN, Yue
单位: PAT联盟
时间限制: 400ms
内存限制: 64MB
代码长度限制: 16KB

代码：

n = input().split()
a = int(n[0])
if a % 2 == 0:
    b = a / 2
else:
    b = (a + 1) / 2
for i in range(int(b)):
    if i == 0 or i == b - 1:
        print(n[1] * a)
    else:
        print('{}{}{}'.format(n[1], ' ' * (a - 2), n[1]))