[swift] 백준 - 2828번: 사과 담기 게임

2828번: 사과 담기 게임

---

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

나의 풀이

let input = readLine()!.split(separator: " ").map { Int($0)! }
let appleNumber = Int(readLine()!)!
let appleLocation = (0..<appleNumber).map { _ in Int(readLine()!)! }
var basket = (1...input[1]).map { $0 }
var move = 0
for i in appleLocation {
    guard !basket.contains(i) else { continue }
    if basket[basket.endIndex-1] < i {
        let distance = i - basket[basket.endIndex-1]
        basket = basket.map { $0 + distance }
        move += distance
        continue
    } else {
        let distance = basket[basket.startIndex] - i
        basket = basket.map { $0 - distance }
        move += distance
        continue
    }
}
print(move)

 

최소한의 움직임으로 떨어지는 사과를 바구니로 받아내면 되는데요.

저는 바구니를 배열로 만들었습니다.

 

바구니 크기가 3이라면

[1,2,3]으로 시작하는 거죠

 

사과가 떨어지는 위치가

바구니 안에 포함된다면

반복문은 guard에 걸려 continue를 하게 됩니다.

 

바구니보다 오른쪽에 있는 곳에서 사과가 떨어진다면

(사과가 떨어지는 위치 - 바구니의 제일 큰 값)로

움직인 횟수를 구하고

움직인 횟수만큼 배열 각각의 값을 더해줍니다.

 

바구니보다 왼쪽에 있는 곳에서 사과가 떨어진다면

(바구니의 제일 작은 값 - 사과가 떨어지는 위치)로

움직인 횟수를 구하고

움직인 횟수만큼 배열 각각의 값을 빼줍니다.