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

박상준·2023년 6월 26일
0

코딩테스트 연습

목록 보기
5/7

Solution

import java.util.Arrays;

// 배열 회전시키기
public class Programmers02 {
    public static int[] solution(int[] numbers, String direction) {
        int[] answer = new int[numbers.length];

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

        return answer;
    }

    public static void main(String[] args) {
        int[] numbers = {1, 2, 3};
        String direction = "right";
        System.out.println(Arrays.toString(solution(numbers, direction)));

        numbers = new int[]{4, 455, 6, 4, -1, 45, 6};
        direction = "left";
        System.out.println(Arrays.toString(solution(numbers, direction)));
    }
}
profile
신입 개발자를 꿈꾸는 코린이

0개의 댓글