[프로그래머스/Kotlin] 공원 산책 (Lv. 1)

문제

지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

• ["방향 거리", "방향 거리" … ]

예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.

• 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
• 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.

공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

 

 

제한 사항

• 3 ≤ park의 길이 ≤ 50
  • 3 ≤ park[i]의 길이 ≤ 50
    • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
      • S : 시작 지점
      • O : 이동 가능한 통로
      • X : 장애물
  • park는 직사각형 모양입니다.
• 1 ≤ routes의 길이 ≤ 50
  • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
  • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
  • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
    • op는 다음 네 가지중 하나로 이루어져 있습니다.
      • N : 북쪽으로 주어진 칸만큼 이동합니다.
      • S : 남쪽으로 주어진 칸만큼 이동합니다.
      • W : 서쪽으로 주어진 칸만큼 이동합니다.
      • E : 동쪽으로 주어진 칸만큼 이동합니다.
    • 1 ≤ n ≤ 9

 

입출력 예

 

풀이 방법

반복문, 델타배열 두가지 방법으로 풀었다. 반복문으로 풀었을 때 코드가 너무 지저분하고 중복되는 코드가 많아서 이를 줄이고자 델타배열을 사용했다.

풀이 방법은 델타배열 풀이법으로 설명하겠다.

 

1. 델타배열을 생성한다.

var dr = arrayOf(-1,0,1,0) // N(북),E(동),S(남),W(서)
var dc = arrayOf(0,1,0,-1)

 

2. 현재 위치 좌표를 알아야하기 때문에 변수를 생성하고 현재 위치를 구한다.

    동시에 현재 주어진 park는 단순 문자열배열로 되어있기 때문에 2차원 배열을 생성해서 각각 해당하는 좌표에 값을 추        가한다.

var currentRow = 0
var currentCol = 0

var array = Array(park.size, {Array(park[0].length, {""})})
for((index, p) in park.withIndex()){
    for(i in 0 until p.length){
        array[index][i] = p[i].toString()

        if(p[i].toString() == "S"){
            currentRow = index
            currentCol = i
        }
    }
}

 

3. 문제에서 주어진 routes를 돈다. "방향 칸" 으로 주어지기 때문에 공백을 기준으로 split으로 나눴다.

    그리고 split으로 자를 경우 문자열이 되기 때문에 num은 정수형으로 형변환한다.

val (direction, number) = route.split(" ")
var num = number.toInt()

    

4. 1번에서 생성했던 델타배열의 인덱스에 맞게 숫자를 지정해준다.

var dic = 0
if(direction == "E"){
    dic = 1
}else if(direction == "W"){
    dic = 3
}else if(direction == "S"){
    dic = 2
}else{
    dic = 0
}

 

5. num만큼 반복하며 배열의 범위를 넘지 않는지 장애물을 만나지 않았는지 체크해준다.

   만약 배열의 범위를 넘지 않고 장애물을 만나지 않았다면 continue하고,

   그렇지 않다면 check 를 false로 만든다.

   check 사용해야하는 이유는 6번에 계속.

var check = true
var r = 0
var c = 0
for(d in 1..num){
    r = currentRow + dr[dic]*d
    c = currentCol + dc[dic]*d

    if(0<=r && r<array.size && 0<=c && c<array[0].size && array[r][c] != "X"){
        continue
    }else{
        check = false
        break
    }
}

 

6. 만약 5번에서 조건을 만족하지 않아 break가 된다면 for문에서만 멈추기 때문에 check를 하지 않으면 조건을 만족하던        하지 않던 currentRow, currentCol의 값을 변경해버린다.

    그렇기 때문에 위의 조건을 모두 만족했을 때만 현재의 위치를 변경하기 위해 check 변수로 체크해준다.

if(check) {
    currentRow = r
    currentCol = c
}

 

7. 마지막으로 정답 배열에 현재 위치를 담아서 리턴한다.

answer += currentRow
answer += currentCol
return answer

 

코드

1. 반복문

class Solution {
    fun solution(park: Array<String>, routes: Array<String>): IntArray {
        var answer: IntArray = intArrayOf()
        
        var currentRow = 0
        var currentCol = 0
        
        var array = Array(park.size, {Array(park[0].length, {""})})
        for((index, p) in park.withIndex()){
            for(i in 0 until p.length){
                array[index][i] = p[i].toString()
                
                if(p[i].toString() == "S"){
                    currentRow = index
                    currentCol = i
                }
            }
        }
        
        for(route in routes){
            val (direction, number) = route.split(" ")
            var num = number.toInt()
            var check = true
            if(direction == "E"){
                for(i in 1..num){
                    if(currentCol+i < array[0].size && array[currentRow][currentCol+i] != "X"){
                        continue
                    }else{
                        check = false
                        break
                    }
                }
                if(check) currentCol += num
                
            }else if(direction == "W"){
                for(i in 1..num){
                    if(currentCol-i >= 0 && array[currentRow][currentCol-i] != "X"){
                        continue
                    }else{
                        check = false
                        break
                    }
                }
                if(check) currentCol -= num
                
            }else if(direction == "S"){
                for(i in 1..num){
                    if(currentRow+i < array.size && array[currentRow+i][currentCol] != "X"){
                        continue
                    }else{
                        check = false
                        break
                    }
                }
                if(check) currentRow += num
                
            }else{
                for(i in 1..num){
                    if(currentRow-i >= 0 && array[currentRow-i][currentCol] != "X"){
                        continue
                    }else{
                        check = false
                        break
                    }
                }
                if(check) currentRow -= num
            }
            println("$currentRow , $currentCol")
        }
        answer += currentRow
        answer += currentCol
        return answer
    }
}

 

 

2. 델타배열 사용

class Solution {
    fun solution(park: Array<String>, routes: Array<String>): IntArray {
        var answer: IntArray = intArrayOf()
        
        var dr = arrayOf(-1,0,1,0) // N(북),E(동),S(남),W(서)
        var dc = arrayOf(0,1,0,-1)
        
        var currentRow = 0
        var currentCol = 0
        
        var array = Array(park.size, {Array(park[0].length, {""})})
        for((index, p) in park.withIndex()){
            for(i in 0 until p.length){
                array[index][i] = p[i].toString()
                
                if(p[i].toString() == "S"){
                    currentRow = index
                    currentCol = i
                }
            }
        }
        
        for(route in routes){
            val (direction, number) = route.split(" ")
            var num = number.toInt()
            var check = true
            var dic = 0
            
            if(direction == "E"){
                dic = 1
            }else if(direction == "W"){
                dic = 3
            }else if(direction == "S"){
                dic = 2
            }else{
                dic = 0
            }
            
            var r = 0
            var c = 0
            for(d in 1..num){
                r = currentRow + dr[dic]*d
                c = currentCol + dc[dic]*d
                
                if(0<=r && r<array.size && 0<=c && c<array[0].size && array[r][c] != "X"){
                    continue
                }else{
                    check = false
                    break
                }
            }
            
            if(check) {
                currentRow = r
                currentCol = c
            }
        }
        answer += currentRow
        answer += currentCol
        return answer
    }
}

 

 

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

프로그래머스