BOJ 3190 뱀 swift

문제 링크 : https://www.acmicpc.net/problem/3190

 

 

문제 해설

정말 단순하게 구현문제입니다. 다른 언어면 아마 Deque를 구현해서 사용했어야 문제를 풀 수 있을겁니다.

뱀 문제를 풀 때 주의해야할 점은 뱀의 시작위치는 (1, 1)이라는 것입니다.

일반 배열처럼 (0, 0)에서 시작하면 틀리기 때문에 주의하시길 바랍니다.

 

코드

import Foundation

let n = Int(readLine()!)!
let k = Int(readLine()!)!

var apples = [(Int, Int)]()
for _ in 0..<k {
    let apple = readLine()!.split(separator: " ").map{ Int(String($0))! }
    apples.append((apple[0], apple[1]))
}

let l = Int(readLine()!)!
var moves = [(Int, String)]()
for _ in 0..<l {
    let move = readLine()!.split(separator: " ").map{ String($0) }
    moves.append((Int(move[0])!, move[1]))
}

let directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]
var dir = 0
var time = 0
var snake = [(Int, Int)]()
snake.append((1, 1))

while true {
	time += 1
    let cur = snake.first!

    let next = (cur.0 + directions[dir].0, cur.1 + directions[dir].1)
    if !(1...n).contains(next.0) || !(1...n).contains(next.1) || snake.contains(where: {$0 == next}) {
        break
    }

    if apples.contains(where: { $0 == next }) {
        apples.removeAll(where: { $0 == next })
        snake.insert(next, at: 0)
    } else {
        snake.insert(next, at: 0)
        snake.removeLast()
    }
    if !moves.isEmpty {
        if moves.first!.0 == time {
            let changeDir = moves.first!.1
            if changeDir == "L" {
                dir = dir == 0 ? 3 : dir - 1
            } else {
                dir = (dir + 1) % 4
            }
            moves.removeFirst()
        }
    }
}

print(time)