PAT 1036 跟奥巴马一起编程

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

输入格式：

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

输出格式：

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

输入样例：

10 a

输出样例：

aaaaaaaaaa
a        a
a        a
a        a
aaaaaaaaaa

代码：

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();
        String a=sc.next();
        String[][] arr=new String[(n+1)/2][n];
        for(int i=0;i<(n+1)/2;i++){
            for(int j=0;j<n;j++){
                arr[i][j]="0";
            }
        }
        for(int i=0;i<(n+1)/2;i++){
            for(int j=0;j<n;j++){
                if(i==0||i==(n+1)/2-1){
                    arr[i][j]=a;
                }
                if(j==0||j==n-1){
                    arr[i][j]=a;
                }
                if(arr[i][j].equals("0")){
                    arr[i][j]=" ";
                }
            }
        }
        for(int i=0;i<(n+1)/2;i++){
            for(int j=0;j<n;j++){
                System.out.print(arr[i][j]);
            }
            System.out.println();
        }
        sc.close();
    }
}