(Swift) [백준/Boj] 1021번: 회전하는 큐

#문제링크

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

1021번: 회전하는 큐

#난이도

Silver 4

 

#접근 방법 및 중요 포인트

문제에서 3가지 연산을 수행해야한다고 제시하고 있다. 연산들을 코드로 어떻게 구현할지 생각해보면 덱(Deque) 자료구조를 사용하면 된다. (앞, 뒤로 삽입 추출이 가능한 자료구조)

1. 첫번째 원소 추출 : 첫번째 원소 삭제
2. 왼쪽으로 1칸 이동 : 0번 원소 추출해서 맨 뒤에 삽입
3. 오른쪽으로 1칸 이동 : 맨 뒤의 원소 추출해서 0번에 삽입

중요한 포인트는 왼쪽 방향으로 가야 원하는 원소를 빨리 뽑을지, 오른쪽으로 가야 빨리 뽑을지 선택하는 로직이었다.

배열의 길이를 2로 나눈뒤, Target이 왼쪽파트에 속하는지, 오른쪽 파트에 속하는지 분기하여 회전시킬 방향을 정하면 된다.

 

 

#Code (Swift)

var nm = readLine()!.split(separator: " ").map{Int($0)!}
let n = nm[0], m = nm[1]
var targets = Array(readLine()!.split(separator: " ").map{Int($0)!})
var arr = Array(1...n)
var answer = 0

func rotate(_ direction: Int){
    if direction == -1{
        let temp = arr.removeFirst()
        arr.append(temp)
    }else{
        let temp = arr.removeLast()
        arr.insert(temp, at: 0)

    }
}
func find(){
    let target = targets[0]
    let targetIdx = arr.firstIndex(of: target)!
    if target == arr[0] {
        targets.removeFirst()
        arr.removeFirst()
    }else if targetIdx > arr.count/2{
        rotate(1)
        answer += 1
    }else{
        rotate(-1)
        answer += 1
    }
}

while(!targets.isEmpty){
    find()
}

print(answer)