[프로그래머스] 배열 회전시키기

🖊️문제

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

 

 

 

🖊️문제 풀이

class Solution {
    public int[] solution(int[] numbers, String direction) {
        int[] answer = new int[numbers.length];
        if(direction.equals("right")){
            for (int i = 0; i < numbers.length; i++) {
                if(i==numbers.length-1){
                    answer[0] = numbers[i];
                }else{
                    answer[i+1] = numbers[i];
                }
            }
        }else if(direction.equals("left")){
            for (int i = 0; i < numbers.length; i++) {
                if(i== numbers.length-1){
                    answer[i] = numbers[0];
                }else{
                    answer[i] = numbers[i+1];
                }
            }
        }
        return answer;
    }
}