[백준 16435] 스네이크버드 (Swift)

문제

16435번: 스네이크버드

스네이크버드가 공중에 달린 과일을 먹을 때 마다 1만큼 길이가 증가한다.

스네이크버드는 자신의 몸 길이보다 낮거나 같은 높이에 있는 과일만 먹을 수 있다.

스네이크버드의 몸 길이와 과일들의 높이가 주어졌을 때, 스네이크버드가 최대로 늘릴 수 있는 몸의 길이를 구한다.

풀이

과일 배열의 순회가 쉽도록, 주어지는 과일 배열을 정렬한다.

그리고 높이가 낮은 과일 순서부터 현재의 몸 길이와 비교한다.

과일이 몸 길이보다 높이 있어서 먹을 수 없게 되면, 정렬된 과일 배열에서는 이후 비교는 의미가 없으므로 반복문을 탈출한다.

과일을 먹을 수 있는 몸 길이라면 몸 길이에 1을 더하고 다음 비교를 계속 진행한다.

let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
let fruitArr = readLine()!.split(separator: " ").map { Int($0)! }.sorted()
var snake = inputArr[1]
for fruit in fruitArr {
    if snake < fruit {
        break
    }
    snake += 1
}
print(snake)