PAT乙级练习题B1036. 跟奥巴马一起编程

题目描述

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

输入格式：

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

输出格式：

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

输入样例：
10 a
输出样例：
aaaaaaaaaa
a a
a a
a a
aaaaaaaaaa

代码

#include<iostream>
using namespace std;

int main()
{
    int N, M;
    char ch;
    cin >> N >> ch;
    if (N % 2 != 0)
    {
        M = N / 2 + 1;
    }
    else
    {
        M = N / 2;
    }
    for (int i = 0; i < M; ++i)
    {
        if (i == 0 || i == M - 1)
        {
            for (int j = 0; j < N; ++j)
            {
                cout << ch;
            }
            cout << endl;
        }
        else
        {
            for (int j = 0; j < N; ++j)
            {
                if (j == 0 || j == N - 1)
                {
                    cout << ch;
                }
                else
                {
                    cout << " ";
                }
            }
            cout << endl;
        }
    }
    system("pause");
    return 0;
}