주어진 셀의 크기 N에 대해 골뱅이를 돌아간 ㅍ 모양으로 출력하는 함수 구현하기
서준이는 아빠로부터 골뱅이가 들어 있는 상자를 생일 선물로 받았다. 상자 안에는 반시계 방향으로 90도 돌아간 ㅍ자 모양의 골뱅이가 들어있다. 돌아간 ㅍ자 모양은 가로 및 세로로 각각 5개의 셀로 구성되어 있다. 상자에는 정사각형 모양의 셀의 크기를 나타내는 숫자 하나가 적혀있다. 셀의 크기 N이 주어지면 예제 출력과 같은 방식으로 골뱅이 모양을 출력하시오.
# 1. N = 1
@ @
@@@@@
@ @
@@@@@
@ @
# 2. N = 3
@@@ @@@
@@@ @@@
@@@ @@@
@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@
@@@ @@@
@@@ @@@
@@@ @@@
@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@
@@@@@@@@@@@@@@@
@@@ @@@
@@@ @@@
@@@ @@@
@
가 채워지는 부분의 인덱스를 구한다.
@
로 채우기 위해, 셀마다 N번의 반복이 추가적으로 필요하다.
@
또는 ‘ ‘
을 출력한다.
위의 풀이를 Python로 작성하면 아래와 같다.
N = int(input().strip())
filled_cells = [{0, 4}, {0, 1, 2, 3, 4}, {0,4}, {0,1,2,3,4}, {0,4}]
for row in range(5):
for _ in range(N):
for col in range(5):
if col in filled_cells[row]:
print('@' * N, end = '')
else:
print(' ' * N, end = '')
print()