BOJ-1107 리모컨 Swift

문제

수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.

리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.

수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

수빈이가 지금 보고 있는 채널은 100번이다.

입력

첫째 줄에 수빈이가 이동하려고 하는 채널 N (0 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 고장난 버튼의 개수 M (0 ≤ M ≤ 10)이 주어진다. 고장난 버튼이 있는 경우에는 셋째 줄에는 고장난 버튼이 주어지며, 같은 버튼이 여러 번 주어지는 경우는 없다.

 

출력

첫째 줄에 채널 N으로 이동하기 위해 버튼을 최소 몇 번 눌러야 하는지를 출력한다.

내가 푼 풀이

-처음 문제를 접했을때, 문제를 어떻게 풀까에 대한 해답이 쉽게 나오지않았다.

-가장 오래걸리는 완전탐색을 배제하고 더 빠른 탐색방법이 없을까 dp? dfs? 여러가지 구성해보다가 결국엔 다른 사람의 풀이 도움을 받아완전탐색으로 풀었다.

-완전탐색도 모든 경우의 수를 탐색해야하므로 탐색기준과 범위를 정하기가 쉽지않았다.

-여기서 완전탐색으로 푼 방법은 아래와 같다.

 

N번의 채널로 이동하는방법

1. 고장나지않은 번호를 입력해 이동한다.

2. +/-버튼을 이용해 한칸씩 이동한다.

3. 고장나지않은 버튼을 이용해 N번채널과 가까운채널로 이동 후 2번방법으로 이동한다.

 

N의 최댓값은 50만이지만 채널은 무한대이다.

현재 채널 100을 기준으로 50만채널로 이동할 때, 단순히 +/-로 이동한다면

100 -> 500,000 보다 990,000 -> 500,000 이동이 훨씬 적게 누른다.

 

100 채널에서 500,000로 이동하려면 +/- 몇번을 눌러야할까

바로 |500,000 - 100|번 눌러야한다.

마찬가지로 990,000에서 500,000채널로 이동하려면 |500,000 - 990,000| 번 누르면 된다.

 

결국 우리가 이동해야할 N 채널은 50만까지 정해져있지만,

최소로 이동하려면 우리는 100만까지 범위의 채널에서 이동해서 최솟값을 찾는게 해답이 될 것이다.

 

이것을 이렇게 생각해냈다.

채널 0부터 100만까지 고장나지 않은 번호로 이동을 했다면,

채널을 이동할때의 누른 횟수와 이동한 채널에서 N채널까지 +/-이동만으로 누른 횟수를 더한다.

더한 값들의 최솟값은 문제에서 구하는 최솟값이 될 것이다.

 

이렇게 생각한것도 다른사람의 풀이 도움을 받아서 생각해낼 수 있었다.

너무 어려웠다

import Foundation

// 입력받기
let N = Int(readLine()!)!
let M = Int(readLine()!)!
var brokenBtns = [Int]()
if M != 0 {
    brokenBtns = readLine()!.split(separator: " ").map{Int(String($0))!}
}
// 현재채널 100, 100에서 누른횟수 최솟값
var minCount = abs(N - 100)

// 100만채널까지 모든 경우의 수를 탐색한다.
for i in 0..<1000001 {
    let len = length(num: i)
    // 고장나지않은 번호로 이동했다면 이동할때 누른 횟수와, 이동한 채널에서 +/-로 이동한 횟수를 더한다.
    if len > 0 {
        let count = abs(N - i)
        minCount = min(minCount, count + len)
    }
}

// 고장나지 않은 번호로 이동했을때 버튼을 누른 횟수를 구한다.
func length(num: Int) -> Int{
    var n = num
    // 목적지 N이 0채널이라면
    if n == 0 {
        // 0번이 고장났다면 번호눌러서 이동할 수 없다.
        // 1번에서 이동해야함.
        if brokenBtns.contains(n)    {
            return 0
        } else {
            return 1
        }
    }
    
    var len = 0
    // 누른횟수 계산
    // 이동하려는 채널에 고장난번호가 포함되있다면 애초에 이동이 불가능하므로 0 리턴
    while n > 0 {
        if brokenBtns.contains(n % 10) { return 0 }
        n = n / 10
        len += 1
    }
    return len
}


print(minCount)