[Lv. 0] 배열 회전시키기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120844

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다. 배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을 return하도록 solution 함수를 완성해주세요.

 

제한 사항 : 

• 3 ≤ numbers의 길이 ≤ 20
• direction은 "left" 와 "right" 둘 중 하나입니다.

입출력 예 :

numbers	direction	result
[1, 2, 3]	"right"	[3, 1, 2]
[4, 455, 6, 4, -1, 45, 6]	"left"	[455, 6, 4, -1, 45, 6, 4]

풀이 : 

class Solution {
    fun solution(numbers: IntArray, direction: String): IntArray {
        var answer: IntArray
        val temp = numbers.toMutableList()

        answer = if (direction == "left") {
            temp.removeAt(0)
            temp.add(numbers[0])
            temp.toIntArray()
        } else {
            temp.add(0, numbers[numbers.lastIndex])
            temp.removeAt(temp.lastIndex)
            temp.toIntArray()
        }

        return answer
        }
}