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

LeeSeungEun·2023년 5월 23일

Coding Test

목록 보기

31/38

1. 문제

2. 코드

import java.util.Arrays;

class Solution {
    public int[] solution(int[] numbers, String direction) {
        int[] answer = new int[numbers.length]; // numbers배열과 같은 크기로 잡아줌

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