14일차)알고리즘 문제(공원 산책), 코드카타 리뷰(indices)

>알고리즘 문제

 

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

2. 솔루션

class Solution {
    fun solution(park: Array<String>, routes: Array<String>): IntArray {
        var current=arrayOf(-1,-1)
        
        //시작위치
        for(i:Int in 0..park.size-1){
            for(j:Int in 0..park[0].length-1){
                if(park[i][j]=='S'){
                    current[0]=i
                    current[1]=j
                    break
                }
            }
            //배열은 참조타입이므로 ==나 !=를 쓰면 참조가 비교됨
            if(!current.contentEquals(arrayOf(-1,-1))){
                break
            }
        }
        
        
        for(i in routes){
            val tmp=i.split(" ")
            val dir=tmp[0]
            val len=tmp[1].toInt()
            val cur=current.copyOf()
            var flag=true
            var idx=1
            
            if(dir=="E"){
                var parkLen=park[0].length
                while(idx++<=len){
                    ++cur[1]
                    if(cur[1]>=parkLen){
                        flag=false
                        break
                    }else if(park[cur[0]][cur[1]]=='X'){
                        flag=false
                        break
                    }
                    
                }
            }
            
            if(dir=="W"){
                while(idx++<=len){
                    --cur[1]
                    if(cur[1]<0){
                        flag=false
                        break
                    }else if(park[cur[0]][cur[1]]=='X'){
                        flag=false
                        break
                    }
                    
                }
            }
            
            if(dir=="S"){
                var parkLen=park.size
                while(idx++<=len){
                    ++cur[0]
                    if(cur[0]>=parkLen){
                        flag=false
                        break
                    }
                    else if(park[cur[0]][cur[1]]=='X'){
                        flag=false
                        break
                    }
                    
                }
            }
            
            if(dir=="N"){
                while(idx++<=len){
                    --cur[0]
                    if(cur[0]<0){
                        flag=false
                        break
                    }else if(park[cur[0]][cur[1]]=='X'){
                        flag=false
                        break
                    }
                    
                }
            }
            if(flag){
                current=cur
            }
        }
        return current.toIntArray()
    }
}

-park의 전체를 검사해 시작지점이 나올때까지 for문을 돌림. 찾으면 break로 탈출

-routes의 각 원소를 for으로 돌면서 어느방향으로 얼마나 이동할지 확인

-flag로 해당 명령을 수행할지 여부 결정. while문으로 이동해보다가 범위를 벗어나거나 장애물을 만나면 flag를 false로 바꿔 해당 명령을 스킵하도록 함.

@배열 비교는 contentEquals()로 하기

 

 

>코드카타 리뷰

1. indices: for문같은걸 돌릴때 컬렉션의 인덱스 범위값을 .size-1 을 쓸 필요없이 indices를 사용하면 된다.

2. withIndex: (index, value) 값을 얻을 수 있음 => for((index,value) in list.withIndex){}

3. return에 when이나 if문을 expression으로 사용할때: 결과값 부분에 return을 추가로 사용가능

=>when이나 if내에 while같은 것을 사용할시 조기종료를 위해 return을 사용하기도 함.

 

@실습 

if(character is Archer) {
    character?.run {
        money += 10000
    }
} else if(character is Wizard) {
    character?.run {
        money += 10000
    }
}

//위의 식을 아래방식으로 개선

character?.let {
    (it as? Archer)?.run { money+=10000}
    (it as? Wizard)?.run { money+=10000}
}

-if문으로 나뉜것을 ?.을 이용하면 어차피 해당 안되는 부분은 null이 반환되며 적용이 안되기 때문에 이런식으로 가독성 높게 줄일 수 있음