BOJ 18242 - 네모네모 시력검사

문제

백준 온라인 저지 - 18242번

풀이 과정

네모네모 안과의 시력 검사를 통과할 수 있는 기능을 구현하는 문제입니다.

격자에는 단 하나의 직사각형이 존재한다는 보장이 있기 때문에
우리가 활용할 수 있는 첫번째 정보는 왼쪽 상단 모서리의 위치입니다.

이후 오른쪽 상단 모서리와 왼쪽 하단 모서리의 위치를 구해서 직사각형의 너비와 높이를 구하고
이를 활용해 직사각형의 각 변을 검사하면 됩니다.

코드

import sys

N, M = list(map(int, sys.stdin.readline().split()))
board = [list(sys.stdin.readline().strip()) for _ in range(N)]


def get_pivot(board):
    pivot = None

    for x in range(N):
        for y in range(M):
            if board[x][y] == '#':
                pivot = [x, y]
                break
        if pivot is not None:
            break

    return pivot


def get_width_height(board, sx, sy):
    right_top, bottom_left = [], []

    # 오른쪽 상단 모서리 찾기
    for y in range(M - 1, sy, -1):
        if board[sx][y] == '#':
            right_top = [sx, y]
            break

    # 왼쪽 하단 모서리 찾기
    for x in range(N - 1, sx, -1):
        if board[x][sy] == '#':
            bottom_left = [x, sy]
            break

    width = right_top[1] - sy + 1
    height = bottom_left[0] - sx + 1

    return width, height


def inspect_board(board, width, height, sx, sy):
    # UP, DOWN
    for y in range(sy, sy + width):
        if board[sx][y] == '.':
            return 'UP'
        if board[sx + height - 1][y] == '.':
            return 'DOWN'

    # LEFT, RIGHT
    for x in range(sx, sx + height):
        if board[x][sy] == '.':
            return 'LEFT'
        if board[x][sy + width - 1] == '.':
            return 'RIGHT'


def solution():
    pivot = get_pivot(board)
    width, height = get_width_height(board, *pivot)
    answer = inspect_board(board, width, height, *pivot)

    return answer


print(solution())