백준 - 톱니바퀴 2(Swift)

백준 - 톱니바퀴 2(Swift)

 

 

 문제 설명

 

https://www.acmicpc.net/problem/15662

15662번: 톱니바퀴 (2)

 

 나의 풀이

 

보통 문제가 굉장히 길면서 입력값이 1000이하라면 구현문제다.

 

이 문제의 해결점은 톱니바퀴가 맞물리는 타이밍이다. 톱니바퀴가 맞물리는 공간은 인덱스2와 6이다.

예시로 4개의 톱니바퀴가 있다고 가정해보자.

2번 톱니바퀴를 시계방향으로 돌릴때 양쪽의 톱니바퀴랑 맞물리는 부분이 확인해야한다. 즉 1번의 6번인덱스와 3번의 2번인덱스를 확인해서 서로 극이 다르다면 재귀를 돌고 아니라면 넘기면 된다. 

 

풀이가 조금 오래걸렸던 이유는 시계방향으로 돌때 맨마지막값을 첫번째에 넣어야하는데 이 부분을 반대로했다. 그리고 방문여부를 체크할때 인덱스가 넘어간상태에서 배열을 서브스크립팅하다보니 런타임에러가 나왔었는데 방문표시를 하기전에 숫자의 유효성을 먼저 체크하니 해결할 수 있었다.

 

import Foundation

struct Wheel {
    var wheel: [Int]

    init(wheel: [Int]) {
        self.wheel = wheel
    }
}


let T = Int(readLine()!)!
var wheels = [Wheel]()
var visited = [Bool](repeating: false, count: T+1)
for i in 0..<T {
    let wheel = Wheel(wheel: readLine()!.map{Int(String($0))!})
    wheels.append(wheel)
}

let K = Int(readLine()!)!
for _ in 0..<K {
    visited = [Bool](repeating: false, count: T+1)
    let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
    let target = input[0], dir = input[1] 
    rotate(num: target, dir: dir)
}
var cnt = 0
wheels.forEach { 
    if $0.wheel[0] == 1 {
        cnt += 1
    }
}
print(cnt)

func rotate(num: Int, dir: Int) {
    visited[num] = true
    // 0보다 크면서 방문되지 않았을 때
    if 0<num-1, !visited[num-1] {
    	// 바퀴가 맞물릴 때
        if wheels[num-2].wheel[2] != wheels[num-1].wheel[6] {
            if dir == 1 { rotate(num: num-1, dir: -1)} 
            else { rotate(num: num-1, dir: 1)}
        }
    }
    if num+1<T+1, !visited[num+1] {
        if wheels[num].wheel[6] != wheels[num-1].wheel[2] {
            if dir == 1 { rotate(num: num+1, dir: -1) }
            else { rotate(num: num+1, dir: 1)}
        }
    }
    
    if dir == 1 { // 시계방향으로 회전
        let tmp = wheels[num-1].wheel.removeLast()
        wheels[num-1].wheel.insert(tmp, at: 0)
    } else {
        wheels[num-1].wheel.append(wheels[num-1].wheel.removeFirst())
    }
}