[프로그래머스] 바탕화면 정리 Python

문제

[프로그래머스] 바탕화면 정리

프로그래머스

 

정답

def solution(wallpaper):
    answer = []
    lr = [100,0]
    ud = [100,0]
    for i in range(0, len(wallpaper)):
        for j in range(0, len(wallpaper[0])):
            if wallpaper[i][j] == '#':
                if j < lr[0]:
                    lr[0] = j
                if j + 1 > lr[1]:
                    lr[1] = j + 1
                if i < ud[0]:
                    ud[0] = i
                if i + 1 > ud[1]:
                    ud[1] = i + 1

    answer.append(ud[0])
    answer.append(lr[0])
    answer.append(ud[1])
    answer.append(lr[1])
    
    return answer

파일 위치(#)가 나오면 왼쪽과 오른쪽인 lr[]와 위와 아래인 ud[]에 각각 최소값과 최대값을 저장한 뒤에 이를 통하여 범위를 구했습니다.