[Swift] 프로그래머스 입문 문제, 배열 회전시키기

문제

 

풀이 방법

1. 인자로 받은 numbers는 상수이기 때문에 변수인 result로 변수 초기화
2. direction이 left일때는 배열의 첫번째 원소를 추가하고 첫번째 원소를 삭제
3. direction이 right일때는 배열의 마지막 원소를 첫번째 원소에 insert하고 마지막 원소를 삭제

 

코드

import Foundation

func solution(_ numbers:[Int], _ direction:String) -> [Int] {
    
    var result: [Int] = numbers
    
    switch direction {
    case "left":
        result.append(result[0])
        result.remove(at: 0)
        break
    case "right":
        result.insert(result[result.count - 1], at: 0)
        result.remove(at: result.count - 1)
        break
    default:
        break
    }
    
    return result
}