빙하가 깨지면서 스노우타운에 떠내려 온"죠르디"는 인생 2막을 위해 주택 건축사업에 뛰어들기로 결심하였습니다. "죠르디"는기둥과 보를 이용하여 벽면 구조물을 자동으로 세우는 로봇을 개발할 계획인데, 그에 앞서 로봇의 동작을 시뮬레이션 할 수 있는 프로그램을 만들고 있습니다. 프로그램은2차원 가상 벽면에 기둥과 보를 이용한 구조물을 설치할 수 있는데, 기둥과 보는길이가 1인 선분으로 표현되며 다음과 같은 규칙을 가지고 있습니다.
기둥은 바닥 위에 있거나 보의 한쪽 끝 부분 위에 있거나, 또는 다른 기둥 위에 있어야 합니다.
보는 한쪽 끝 부분이 기둥 위에 있거나, 또는 양쪽 끝 부분이 다른 보와 동시에 연결되어 있어야 합니다.
단, 바닥은 벽면의 맨 아래 지면을 말합니다.
2차원 벽면은n x n크기 정사각 격자 형태이며, 각 격자는1 x 1크기입니다. 맨 처음 벽면은 비어있는 상태입니다. 기둥과 보는 격자선의 교차점에 걸치지 않고, 격자 칸의 각 변에 정확히 일치하도록 설치할 수 있습니다. 다음은 기둥과 보를 설치해 구조물을 만든 예시입니다.
예를 들어, 위 그림은 다음 순서에 따라 구조물을 만들었습니다.
(1, 0)에서 위쪽으로 기둥을 하나 설치 후, (1, 1)에서 오른쪽으로 보를 하나 만듭니다.
(2, 1)에서 위쪽으로 기둥을 하나 설치 후, (2, 2)에서 오른쪽으로 보를 하나 만듭니다.
(5, 0)에서 위쪽으로 기둥을 하나 설치 후, (5, 1)에서 위쪽으로 기둥을 하나 더 설치합니다.
(4, 2)에서 오른쪽으로 보를 설치 후, (3, 2)에서 오른쪽으로 보를 설치합니다.
만약 (4, 2)에서 오른쪽으로 보를 먼저 설치하지 않고, (3, 2)에서 오른쪽으로 보를 설치하려 한다면 2번 규칙에 맞지 않으므로 설치가 되지 않습니다. 기둥과 보를 삭제하는 기능도 있는데 기둥과 보를 삭제한 후에 남은 기둥과 보들 또한 위 규칙을 만족해야 합니다. 만약, 작업을 수행한 결과가 조건을 만족하지 않는다면 해당 작업은 무시됩니다.
벽면의 크기 n, 기둥과 보를 설치하거나 삭제하는 작업이 순서대로 담긴 2차원 배열 build_frame이 매개변수로 주어질 때, 모든 명령어를 수행한 후 구조물의 상태를 return 하도록 solution 함수를 완성해주세요.
국어 지문 문제. 구현 문제는 이런 식으로 문제에 대한 needs와 그에 따라 설정되는 조건에 맞춰 구현하는 역량이 제일 중요하며, 이 문제는 그 중에서도 훌륭한 문제라고 생각된다.
기둥과 보 규칙을 제대로 이해하고 구현하면 되는 문제로, 어떤 좌표에서 그 기둥이 어디에 영향을 받고 어디에 영향을 미치는가에 대해 적용해보면 간단하다. 다음과 같이 그림으로 나타나면 이해가 더 빠를 것이다.
기둥(a = 0)의 관계 그림
우선 x, y 좌표에서의 기둥부터 생각해 보자. 기둥은 1.초록색으로 표시된 보들 혹은연두색으로 표시된 기둥을 적어도 하나 이상 만족해야 존재할 수 있으며,2.빨간색으로 표시된 보들 그리고 주황섹으로 표시된 기둥에 영향을 미친다.
보(a = 1)의 관계 그림
보의 경우는 다음과 같은 관계를 가진다. 보는 1.초록색으로 표시된 보들 혹은연두색으로 표시된 기둥 중 하나을 적어도 하나 이상 만족해야 존재할 수 있으며, 2. 초록색으로 표시된 보들 그리고 빨간색으로 표시된 기둥들에 영향을 미친다.
즉 구조물을 추가할 때는 그 구조에 따라 1번으로 표시된 좌표를 검사하고, 구조물을 삭제할 때는 2번으로 표시된 좌표들의 구조물을 검사하면 된다.
풀이 코드
pillar_check_list = [(0, 1, 1), (-1, 1, 1), (0, 1, 0)]
beam_check_list = [(0, 0, 0), (1, 0, 0), (-1, 0, 1), (1, 0, 1)]
def solution(n, build_frame):
answer = []
build_list = [[[False]*2 for _ in range(n+1)] for _ in range(n+1)]
def check_vaild(x, y, a):
if a == 0 :
return y == 0 or build_list[y][x][1] or (x > 0 and build_list[y][x-1][1]) or build_list[y-1][x][0]
else :
return build_list[y-1][x][0] or build_list[y-1][x+1][0] or (0 < x and build_list[y][x-1][1] and build_list[y][x+1][1])
for x, y, a, b in build_frame :
if b == 0 : ### remove
flg = True
build_list[y][x][a] = False
check_list = pillar_check_list if a == 0 else beam_check_list
for dx, dy, typ in check_list :
ax, ay = x + dx, y + dy
if -1 < ax <= n and -1 < ay <= n and build_list[ay][ax][typ] :
flg = flg and check_vaild(ax, ay, typ)
if not flg :
build_list[y][x][a] = True
else :
if check_vaild(x, y, a) :
build_list[y][x][a] = True
for x in range(n+1) :
for y in range(n+1) :
for typ in range(2) :
if build_list[y][x][typ] :
answer.append([x, y, typ])
return answer
