Codeup 1356 : 사각형 출력하기 2

 

https://codeup.kr/problem.php?id=1356

 

1356 : 사각형 출력하기 2 

n = int(input())

for i in range(1, n+1):
    if (i == 1) or (i == n):
        for j in range(1, n+1):
            print('*', end = '')
        print()
    else:
        print('*', end = '')
        for j in range(1, n-1):
            print(' ', end = '')
        print('*', end = '')
        print()