프로그래머스 - 방문 길이(Swift)

프로그래머스 - 방문 길이(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 나의 풀이

 

처음에는 출발한 지점에서 도착한 지점을 저장하기 위해서 [(Int, Int): [(Int, (Int)]]형식의 딕셔너리를 선언했다. 하지만 튜플은 Hashable하지 않아서되지 않았기에 문자열: 문자열배열의 형태로 두었다. 

 

좌표를 문자열로 만들고 딕셔너리에 저장하면 상당히 지저분한 코드가 된다. 그래서 다른 사람의 풀이를 참고하니 구조체를 이용해서 좌표들을 저장하는 것을 볼 수 있었다.

 

이 문제의 핵심은 기존좌표의 출발점과 도착점이 새로운 좌표의 출발점과 도착점이 같으면 카운트하지 않다는 것과 그 반대의 경우도 카운트하지 않는다는 것이다.

즉 (5, 5)의 좌표에서 (5, 4)로 위로 한번 이동하는 것과 (5, 4)좌표에서 (5, 5)로 이동하는 것은 중복해서 카운트하지 않는다는 것이다.

 

이 부분만 잘 처리하면 문제는 해결된다. 

 

import Foundation

struct Road {
    // x,y: 원래 있던 곳 nx,ny: 이동한 곳 
    let x: Int
    let y: Int
    let nx: Int
    let ny: Int
}

func contains(_ roads: [Road], _ newRoad: Road) -> Bool {
    for road in roads {
        if road.x == newRoad.x && road.y == newRoad.y && road.nx == newRoad.nx && road.ny == newRoad.ny {
            return true
        }
        
        if road.x == newRoad.nx && road.y == newRoad.ny && road.nx == newRoad.x && road.ny == newRoad.y {
            return true
        }
    }
    
    return false
}

func solution(_ dirs:String) -> Int {
    var x = 5, y = 5
    var roads: [Road] = []
    for dic in dirs {
        let prevX = x, prevY = y
        switch dic {
            case "U": if x-1 >= 0 { x -= 1 }
            case "D": if x+1 <= 10 { x += 1 }
            case "L": if y-1 >= 0 { y -= 1 }
            case "R": if y+1 <= 10 { y += 1 }
            default:
            break
        }
        
        if x == prevX && y == prevY {
            continue
        }
        let road = Road(x: prevX, y: prevY, nx: x, ny: y)
        if !contains(roads, road) {
            roads.append(road)
        }
        
    }
    return roads.count
    
}

 

 

 피드백

 

처음에 나는 주어진 방향들을 탐색하면서 범위값만 체크하고 실제 연산부분은 다른 함수로 넣었다.

하지만 범위값을 체크하면서 연산을하고 그 부분이 유효한지를 테스트하는 것이 훨씬 코드가 줄고 직관적이였다.

 

코드가 너무 길어지는 것이 아니라면 하나의 블럭안에서 되도록 많이 처리하는 것이 좋다고 느꼈다.