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

문제 설명

코딩테스트를 준비하는 머쓱이는 프로그래머스에서 문제를 풀고 나중에 다시 코드를 보면서 공부하려고 작성한 코드를 컴퓨터 바탕화면에 아무 위치에나 저장해 둡니다. 저장한 코드가 많아지면서 머쓱이는 본인의 컴퓨터 바탕화면이 너무 지저분하다고 생각했습니다. 프로그래머스에서 작성했던 코드는 그 문제에 가서 다시 볼 수 있기 때문에 저장해 둔 파일들을 전부 삭제하기로 했습니다.

컴퓨터 바탕화면은 각 칸이 정사각형인 격자판입니다. 이때 컴퓨터 바탕화면의 상태를 나타낸 문자열 배열 wallpaper가 주어집니다. 파일들은 바탕화면의 격자칸에 위치하고 바탕화면의 격자점들은 바탕화면의 가장 왼쪽 위를 (0, 0)으로 시작해 (세로 좌표, 가로 좌표)로 표현합니다. 빈칸은 ".", 파일이 있는 칸은 "#"의 값을 가집니다. 드래그를 하면 파일들을 선택할 수 있고, 선택된 파일들을 삭제할 수 있습니다. 머쓱이는 최소한의 이동거리를 갖는 한 번의 드래그로 모든 파일을 선택해서 한 번에 지우려고 하며 드래그로 파일들을 선택하는 방법은 다음과 같습니다.

예를 들어 wallpaper = [".#...", "..#..", "...#."]인 바탕화면을 그림으로 나타내면 다음과 같습니다.

이러한 바탕화면에서 다음 그림과 같이 S(0, 1)에서 E(3, 4)로 드래그하면 세 개의 파일이 모두 선택되므로 드래그 한 거리 (3 - 0) + (4 - 1) = 6을 최솟값으로 모든 파일을 선택 가능합니다.

문제 풀이

바탕화면: 직사각형의 2차원 배열
ex) [".#...", "..#..", "...#."] 에서 '.'는 빈칸, '#'는 파일

방법

2중 for문으로 2차원 배열 돌면서 조건에 해당하는 X1,Y1,X2,Y2의 값 검색

최소한의 이동 거리를 갖는 시작점(X1, Y1) 과 끝점(X2,Y2) 의 조건은 아래와 같다.

• X1: 가장 위쪽의 원소의 행
• Y1: 가장 왼쪽의 원소의 행
• X2: 가장 아래쪽의 원소의 행
• Y2: 가장 오른쪽의 원소의 행

def solution(wallpaper):
    start_x, start_y, end_x, end_y = 51,51,-1,-1
    answer = []
    
    for row, arr in enumerate(wallpaper):
        for col, char in enumerate(arr):
            if(char == '#'):
                #시작점 설정
                if row < start_x : start_x = row
                if col < start_y : start_y = col
                
                #끝점 설정
                if row > end_x : end_x = row 
                if col > end_y : end_y = col 
            
    answer.append(start_x)
    answer.append(start_y)
    answer.append(end_x + 1)
    answer.append(end_y + 1)
    return answer

후기)
끝점은 드래그 좌표 특징에 의해 end_x, end_y 의 값에 각각 1을 더해줘야 한다는걸 주의해야겠다.