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

Nov 19, 2024

school.programmers.co.kr

https://school.programmers.co.kr/learn/courses/30/lessons/120844

문제 설명

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

제한사항

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;
            }
        }
    }