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

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

프로그래머스

문제

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

 

풀이

numbers 배열을 direction 값에 따라 앞 / 뒤로 한 칸 이동시키는 문제이다.

 

direction이 "right"일 경우와 "left"일 경우를 if - else 문으로 나누어 구현해도 되지만,

필자는 아래 코드처럼 조건 연산자인 ? : 을 사용하여 이동 방향을 설정한 뒤 한 번에 구현해 보았다.

int move = direction.equals("right") ? 1 : -1;

 

코드

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

 

틀린 부분이 있다면 정정해 주시면 감사하겠습니다.
궁금한 부분이 있거나, 다른 아이디어가 있으시면 자유롭게 댓글 남겨주세요!