프로그래머스(Lv1) – 바탕화면 정리 문제(파이썬)

프로그래머스 문제풀이

문제

문제설명

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

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

  • 드래그는 바탕화면의 격자점 S(luxluy)를 마우스 왼쪽 버튼으로 클릭한 상태로 격자점 E(rdxrdy)로 이동한 뒤 마우스 왼쪽 버튼을 떼는 행동입니다. 이때, “점 S에서 점 E로 드래그한다“고 표현하고 점 S와 점 E를 각각 드래그의 시작점, 끝점이라고 표현합니다.
  • 점 S(luxluy)에서 점 E(rdxrdy)로 드래그를 할 때, “드래그 한 거리“는 |rdx – lux| + |rdy – luy|로 정의합니다.
  • 점 S에서 점 E로 드래그를 하면 바탕화면에서 두 격자점을 각각 왼쪽 위, 오른쪽 아래로 하는 직사각형 내부에 있는 모든 파일이 선택됩니다.

예를 들어 wallpaper = [“.#…”, “..#..”, “…#.”]인 바탕화면을 그림으로 나타내면 다음과 같습니다.
eg1.png
이러한 바탕화면에서 다음 그림과 같이 S(0, 1)에서 E(3, 4)로 드래그하면 세 개의 파일이 모두 선택되므로 드래그 한 거리 (3 – 0) + (4 – 1) = 6을 최솟값으로 모든 파일을 선택 가능합니다.
eg1-2.png
(0, 0)에서 (3, 5)로 드래그해도 모든 파일을 선택할 수 있지만 이때 드래그 한 거리는 (3 – 0) + (5 – 0) = 8이고 이전의 방법보다 거리가 늘어납니다.

머쓱이의 컴퓨터 바탕화면의 상태를 나타내는 문자열 배열 wallpaper가 매개변수로 주어질 때 바탕화면의 파일들을 한 번에 삭제하기 위해 최소한의 이동거리를 갖는 드래그의 시작점과 끝점을 담은 정수 배열을 return하는 solution 함수를 작성해 주세요. 드래그의 시작점이 (luxluy), 끝점이 (rdxrdy)라면 정수 배열 [luxluyrdxrdy]를 return하면 됩니다.

제한사항

  • 1 ≤ wallpaper의 길이 ≤ 50
  • 1 ≤ wallpaper[i]의 길이 ≤ 50
    • wallpaper의 모든 원소의 길이는 동일합니다.
  • wallpaper[i][j]는 바탕화면에서 i + 1행 j + 1열에 해당하는 칸의 상태를 나타냅니다.
  • wallpaper[i][j]는 “#” 또는 “.”의 값만 가집니다.
  • 바탕화면에는 적어도 하나의 파일이 있습니다.
  • 드래그 시작점 (luxluy)와 끝점 (rdxrdy)는 lux < rdxluy < rdy를 만족해야 합니다.

입출력 예

wallpaperresult
[“.#…”, “..#..”, “…#.”][0, 1, 3, 4]
[“……….”, “…..#….”, “……##..”, “…##…..”, “….#…..”][1, 3, 5, 8]
[“.##…##.”, “#..#.#..#”, “#…#…#”, “.#…..#.”, “..#…#..”, “…#.#…”, “….#….”][0, 0, 7, 9]
[“..”, “#.”][1, 0, 2, 1]

입출력 예 설명

입출력 예 #1

  • 문제 설명의 예시와 같은 예제입니다. (0, 1)에서 (3, 4)로 드래그 하면 모든 파일을 선택할 수 있고 드래그 한 거리는 6이었고, 6보다 적은 거리로 모든 파일을 선택하는 방법은 없습니다. 따라서 [0, 1, 3, 4]를 return합니다.

입출력 예 #2

  • 예제 2번의 바탕화면은 다음과 같습니다.eg2.png
    (1, 3)에서 (5, 8)로 드래그하면 모든 파일을 선택할 수 있고 이보다 적은 이동거리로 모든 파일을 선택하는 방법은 없습니다. 따라서 가장 적은 이동의 드래그로 모든 파일을 선택하는 방법인 [1, 3, 5, 8]을 return합니다.

입출력 예 #3

  • 예제 3번의 바탕화면은 다음과 같습니다.
    eg3.png
    모든 파일을 선택하기 위해선 바탕화면의 가장 왼쪽 위 (0, 0)에서 가장 오른쪽 아래 (7, 9)로 드래그 해야만 합니다. 따라서 [0, 0, 7, 9]를 return합니다.

입출력 예 #4

  • 예제 4번의 바탕화면은 다음과 같이 2행 1열에만 아이콘이 있습니다.
    eg4.png
    이를 드래그로 선택하기 위해서는 그 칸의 왼쪽 위 (1, 0)에서 오른쪽 아래 (2, 1)로 드래그 하면 됩니다. (1, 0)에서 (2, 2)로 드래그 해도 아이콘을 선택할 수 있지만 이전보다 이동거리가 늘어납니다. 따라서 [1, 0, 2, 1]을 return합니다.

문제해석과정

  1. 이번 문제를 요약하자면 한번의 드래그로 바탕화면에 있는 모든 파일을 삭제하고자 합니다.
  2. 단, 이때 드래그가 최소한의 크기일때의 좌표를 구하는 것입니다.
  3. 그렇기 때문에 바탕화면의 좌,우,상,하 각 모서리의 위치를 확인하면 원하는 출력값을 구하실수 있습니다.

문제 풀이전에 지문과 제한사항을 반드시 읽으시길 바랍니다.

나의 풀이

이번 문제에서는 for문, if문, enumerate() 등의 함수를 사용해서 풀었습니다.

우선, lux와 luy는 드래그의 시작 포인트인 S점의 x,y 좌표이며, 제한사항에서 주어진 최댓값 50을 디폴트 값으로 지정해줍니다.

같은 맥락으로 rdx와 rdy는 드래그의 끝 포인트인 E점의 x,y 좌표이므로 최소위치인 0을 디폴드 값으로 지정했습니다.

제공된 리스트를 통해서 순차적으로 탐색하면서 파일이 있다면 좌표값을 구분하여 저장합니다.

끝으로 도출된 좌표값을 출력하면 됩니다.

def solution(wallpaper):
    lux, luy, rdx, rdy = 50,50,0,0      # 좌표 디폴트값 선언
    
    for i,v in enumerate(wallpaper):    # wallpaper리스트로 반복
        for i2,v2 in enumerate(v):      # wallpaper의 원소로 반복
            if v2 == '#':               # '#'(파일)이 발견될때
                lux = min(i,lux)        # lux와 비교하여 작은 값 저장
                luy = min(i2,luy)       # luy와 비교하여 작은 값 저장
                rdx = max(i,rdx)        # rdx와 비교하여 큰 값 저장
                rdy = max(i2,rdy)       # rdy와 비교하여 큰 값 저장

    return [lux, luy, rdx+1, rdy+1]     # 최종적으로 도출된 좌표 출력(E점은 파일의 끝부분 위치가 필요하므로 x,y값에 각각 +1) 

다른 사람의 풀이

다른 풀이는 a와 b라는 리스트 2개를 만들어서 파일의 각 x 좌표를 a리스트에 저장, 각 y의 좌표를 b리스트에 저장하였네요.

그리고 a 리스트에서 제일 작은 값을 lux, 제일 큰 값을 rdx로 사용하고 b 리스트에서 제일 작은 값을 luy, 제일 큰 값을 rdy로 사용하여 결과를 출력해주었습니다.

def solution(wall):
    a, b = [], []
    for i in range(len(wall)):
        for j in range(len(wall[i])):
            if wall[i][j] == "#":
                a.append(i)
                b.append(j)
    return [min(a), min(b), max(a) + 1, max(b) + 1]

관련 링크

문제해결과정에서 기초가 부족하다고 느껴지시면 아래 정리 글을 참고해주세요.
추가적으로 궁금한 사항이나 글에 대한 개선점 알려주시면 피드백 적극 반영하도록 하겠습니다.


Leave a Comment