[프로그래머스] 기둥과 보 설치 - Python

프로그래머스

POINT

삭제할 수 있는 조건은 복잡하지만 설치할 수 있는 조건은 비교적 쉽게 구현할 수 있습니다. 그러니 우선 삭제/설치한 다음에 매번 가능한 구조물인지 확인하고, 불가능하면 원상복구하도록 풀어봅시다.

기둥을 설치할 수 있는 조건

기둥의 좌표를 (x, y)라고 하면 세 가지 조건 중 하나를 만족하면 됩니다.

기둥 설치 조건

보를 설치할 수 있는 조건

보의 좌표를 (x, y)라고 하면 두 가지 조건 중 하나를 만족하면 됩니다.

보 설치 조건

풀이

# 설치물이 조건을 만족하는지 확인한다
def is_possible(result):
    for x, y, a in result:
        # 기둥이면
        if a == 0:
            if y == 0 or (x, y-1, 0) in result or (x-1, y, 1) in result or (x, y, 1) in result:
                continue
            return False # 조건을 만족하지 못하면 불가능
        
        # 보이면
        if a == 1:
            if (x, y-1, 0) in result or (x+1, y-1, 0) in result or ((x-1, y, 1) in result and (x+1, y, 1) in result):
                continue
            return False # 조건을 만족하지 못하면 불가능
    
    return True 

def solution(n, build_frame):
    result = []
    
    for x, y, a, b in build_frame:
        # 삭제하는 경우
        if b == 0:
            result.remove((x, y, a)) # 우선 삭제
            # 조건에 맞지 않으면 원위치
            if not is_possible(result): # 조건에 맞지 않으면 원위치
                result.append((x, y, a))
                
        # 설치하는 경우
        else:
            result.append((x, y, a)) # 우선 설치
            # 조건에 맞지 않으면 원위치
            if not is_possible(result):
                result.remove((x, y, a))
    
    return sorted(result)

풀이출처

이것이 취업을 위한 코딩 테스트다 with 파이썬 - YES24