백준 - 컨베이어 벨트 위의 로봇(Swift)

백준 - 컨베이어 벨트 위의 로봇(Swift)

 

 

 문제 설명

 

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

20055번: 컨베이어 벨트 위의 로봇

 

 

 나의 풀이

 

문제를 꼼꼼하게 읽지 못해서 시간이 오래걸렸다.

 

다른 블로그를 보니 나처럼 문제 이해에 시간을 많이 쓴 사람들이 많은 것 같다. 

컨베이어 벨트가 회전하면서 N에서 로봇을 내리는 건데, 아래쪽에서도 로봇이 있는 줄 알고 이를 고려해서 문제를 풀었었다. 

그리고 로봇이 컨베이어벨트와 함께 회전을 할 때는 내구도가 감소하지 않는다. 문제에서 로봇이 "이동"할때만 내구도가 감소한다고 적혀있기 때문이다. 

마지막으로 단계는 4가지 과정 중 첫번째 과정을 시작함과 동시에 결정되기 때문에 result변수를 처음에 증가시키게 했다.

 

이렇게 모든 경우를 적용시켰는데 시간초과가 나와서 다시한번 코드를 확인해보니 첫 번째 과정인 컨베이어 벨트를 회전시켰을때 N에 해당하는 칸의 로봇을 바로 내려줘야 했는데 내려주지 않은 것이 문제가 되었다.

 

 

 import Foundation

struct Belt {
    var robot: Bool // 로봇이 있는지
    var duration: Int // 내구도
}

let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
let N = input[0], K = input[1] // 벨트 칸 수, 내구도 0의 수용치
let arr = readLine()!.components(separatedBy: " ").map{Int($0)!}
var belts = [Belt]()
for (i, x) in arr.enumerated() {
    belts.append(Belt(robot: false, duration: x))
}
var result = 0

while true {
    result += 1
    // 1
    belts.insert(belts.removeLast(), at: 0) // 맨 마지막 벨트를 앞으로
    if belts[N-1].robot == true { belts[N-1].robot = false }
    // 2
    for i in stride(from: N-2, to: -1, by: -1) { // 로봇의 이동
        if belts[i].robot == true, belts[i+1].robot == false, belts[i+1].duration > 0 {
            belts[i].robot = false
            belts[i+1].robot = true
            belts[i+1].duration -= 1
            if i+1 == N-1 { belts[i+1].robot = false } // N인덱스에 로봇이 위치하면 바로 내려준다.
        }
    }
    // 3
    if belts[0].robot == false && belts[0].duration > 0 {
        belts[0].duration -= 1
        belts[0].robot = true
    }
    // 4
    let zeroCount = belts.filter{$0.duration == 0}.count
    if zeroCount >= K {
        break
    }
}
print(result)

 

 

 피드백

 

구현문제는 항상 그렇듯 문제를 정확하게 이해하고 꼼꼼하게 구현하는 것이 중요하다. 

이 문제를 풀기전에 내구도를 위한 배열과 각 벨트마다 id를 주려고했는데 이와 같은 부분들은 필요가 없었다. 

 

아직 많이 부족한 것 같다.