[프로그래머스/Kotlin] 바탕화면 정리 (Lv. 1)

문제

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

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

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

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

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

(0, 0)에서 (3, 5)로 드래그해도 모든 파일을 선택할 수 있지만 이때 드래그 한 거리는 (3 - 0) + (5 - 0) = 8이고 이전의 방법보다 거리가 늘어납니다.

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

 

제한사항

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

 

입출력 예

 

 

풀이 방법

1. 내 풀이법

#을 모두 포함한 최소한의 직사각형 크기를 구하면 되기 때문에 위, 아래, 왼쪽, 오른쪽 총 4번의 반복문을 돌면서 처음 #을 만나는 인덱스를 반환하는 풀이를 생각했다.

 

1. 위의 풀이를 풀기 위해서는 현재는 왼쪽에서 오른쪽 방향인 배열이 주어져 있는데 위에서 아래 방향인 배열도 추가로 있으면 편하겠다고 생각해서 wallpaper2를 만들었다.

var wallpaper2 = Array(wallpaper[0].length, {""})

for(i in 0 until wallpaper[0].length){
    for(j in 0 until wallpaper.size){
        wallpaper2[i] += wallpaper[j][i].toString()
    }
}

 

2. 위에서 아래 방향으로 반복문을 돌면서 #을 만나면 answer에 해당 인덱스를 추가하고 멈춘다.

for(i in 0 until wallpaper.size){
    if(wallpaper[i].contains("#")){
        answer += i
        break
    }
}

 

3. 왼쪽에서 오른쪽 방향으로 반복문을 돌면서 #을 만나면 answer에 해당 인덱스를 추가하고 멈춘다.

for(i in 0 until wallpaper2.size){
    if(wallpaper2[i].contains("#")){
        answer += i
        break
    }
}

 

4. 아래에서 위 방향으로 반복문을 돌면서 #을 만나면 answer에 해당 인덱스 + 1을 추가하고 멈춘다.

1을 더해주는 이유는 만약 아래 사진과 같이 아래에서 위 방향으로 갈 때 2번째 인덱스에서 #을 만난다. 이때 보다시피 3까지 드래그를 해줘야 포함되기 때문에 1을 더해줘야한다.

for(i in wallpaper.size-1 downTo 0){
    if(wallpaper[i].contains("#")){
        answer += i+1
        break
    }
}

 

5. 오른쪽에서 왼쪽 방향으로 반복문을 돌면서 #을 만나면 answer에 해당 인덱스 + 1을 추가하고 멈춘다.

4번과 같은 이유로 1을 더해준다.

for(i in wallpaper2.size-1 downTo 0){
    if(wallpaper2[i].contains("#")){
        answer += i+1
        break
    }
}

 

 

생각보다 시간이 오래 걸렸고 이렇게 전체 방향을 일일히 체크해주는거 말고 한 번에 할 수 있는 방법이 있지 않을까? 하고 다른 사람의 풀이를 봤는데 너무 혁신적인 풀이가 있었다.

 

2. 다른 사람 풀이법

1. x와 y의 최소, 최대 값을 초기화한다.

var minX = wallpaper.size
var minY = wallpaper[0].length
var maxX = 0
var maxY = 0

 

2. 직사각형을 한칸 씩 방문하면서 각각 minX(왼쪽) 과 minY(위)에는 최소값으로 갱신하고 maxX(오른쪽)과 maxY(아래)에는 최대값으로 갱신한다.

for (i in wallpaper.indices) {
    for (j in wallpaper[i].indices) {
        if (wallpaper[i][j] == '#') {
            minX = min(i, minX)
            minY = min(j, minY)
            maxX = max(i, maxX)
            maxY = max(j, maxY)
        }
    }
}

 

3. 값을 리턴한다.

return intArrayOf(minX, minY, maxX + 1, maxY + 1)

진짜 이 코드를 보고 이마를 탁 쳤다.

시간도 훨씬 짧게 나온다. 이런 생각은 어떻게 해내는걸까 ... 내가 너무 어렵게 생각하는건가 아님 내 생각이 단순한건가

열심히 많이 풀고 다른 사람 풀이도 자주 보면 나도 이런 기발한 생각을 할 수 있겠징 ???!!! ㅎ

 

 

 

코드

1. 내 풀이

class Solution {
    fun solution(wallpaper: Array<String>): IntArray {
        var answer: IntArray = intArrayOf()
        
        var wallpaper2 = Array(wallpaper[0].length, {""})
        
        
        for(i in 0 until wallpaper[0].length){
            for(j in 0 until wallpaper.size){
                wallpaper2[i] += wallpaper[j][i].toString()
            }
        }
        
        
        //위
        for(i in 0 until wallpaper.size){
            if(wallpaper[i].contains("#")){
                answer += i
                break
            }
        }
        
        //왼쪽
        for(i in 0 until wallpaper2.size){
            if(wallpaper2[i].contains("#")){
                answer += i
                break
            }
        }
        
        //아래
        for(i in wallpaper.size-1 downTo 0){
            if(wallpaper[i].contains("#")){
                answer += i+1
                break
            }
        }
        
        //오른쪽
        for(i in wallpaper2.size-1 downTo 0){
            if(wallpaper2[i].contains("#")){
                answer += i+1
                break
            }
        }
        
        
        return answer
    }
}

 

2. 다른 사람 풀이

import kotlin.math.*

class Solution {
    fun solution(wallpaper: Array<String>): IntArray {
        var minX = wallpaper.size
        var minY = wallpaper[0].length
        var maxX = 0
        var maxY = 0

        for (i in wallpaper.indices) {
            for (j in wallpaper[i].indices) {
                if (wallpaper[i][j] == '#') {
                    minX = min(i, minX)
                    minY = min(j, minY)
                    maxX = max(i, maxX)
                    maxY = max(j, maxY)
                }
            }
        }
        return intArrayOf(minX, minY, maxX + 1, maxY + 1)
    }
}

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/161990

프로그래머스