이코테 유형별 기출문제

구현

기둥과 보 설치

풀이특징

• 1000개의 데이터 확인 -> 1000번 각각 수행시 현재까지의 구조물 상태를 완전탐색으로 확인이 가능
• 따라서 create 의 경우 넣고 알맞는 구조물인지 확인
• 따라서 delete 의 경우 제거하고 알맞는 구조물인지 확인 하는 식으로 가능
• 알맞는 구조물인지 확인하는 함수 필요
• 각 구조물 상태를 체크할 때 구조물들 중 필연적으로 존재해야 하는 구조물이 있는지 탐색과정이 필요
• in 메소드를 통해 배열 통으로 존재하는지 확인 가능

# 최종 구조물의 상태를 x좌표 오름차순 -> y좌표 오름차순 -> 기둥 -> 보 순으로 정렬 [x, y, a]
# a: 0=기둥, 1=보
# b: 0=삭제, 1=설치

# 현재 설치된 구조물이 가능한 구조물인지 확인
def possible(answers):
    STICK = 0
    TABLE = 1
    for x, y, obj in answers:
        # 기둥인 경우
        if obj == STICK:
            # 바닥 위인 경우
            if y == 0:
                continue
            # 보의 좌측 끝 위인 경우
            elif [x-1, y, 1] in answers:
                continue
            # 보의 우측 끝 위인 경우
            elif [x, y, 1] in answers:
                continue
            # 기둥 위인 경우
            elif [x, y-1, 0] in answers:
                continue
            else:
                return False
        # 보인 경우
        elif obj == TABLE:
            # 좌측 끝이 기둥 위인 경우
            if [x, y-1, 0] in answers:
                continue
            # 우측 끝이 기둥 위인 경우
            elif [x+1, y-1, 0] in answers:
                continue
            # 양끝이 보인 경우
            elif ([x-1, y, 1] in answers and [x+1, y, 1] in answers):
                continue
            else:
                return False
    return True

def solution(n, build_frame):
    DELETE = 0
    CREATE = 1
    # 구조물 전체
    answers = []
    for frame in build_frame:
        x, y, obj, operate = frame
        if operate == DELETE:
            answers.remove([x, y, obj])
            if not possible(answers):
                answers.append([x, y, obj])
        elif operate == CREATE:
            answers.append([x, y, obj])
            if not possible(answers):
                answers.remove([x, y, obj])

    answers.sort()
    return answers