[Java 알고리즘 문제] (9) 배열 회전시키기

문제 설명

제한사항

• 3 ≤ numbers의 길이 ≤ 20

• direction은 "left" 와 "right" 둘 중 하나입니다.

해답

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