프로그래머스 - 배열 회전시키기 / JAVA (자바) 코딩 알고리즘 연습

반응형

풀이는 간단하다.

​

if 문으로 (switch case 문도 가능하다.) direction 파라미터가 left 일 때와 right 일 때로 구분하여 whlie 문을 돌려서 새로운 answer 배열에 넣어서 출력하면 된다.

​

익숙한 if문 대신에 while 문을 사용했더니 처리 속도가 빨라지니 앞으로는 처리속도도 신경쓰면서 로직을 짜려고 노력해보자~~!!

 

class Solution {
    public int[] solution(int[] numbers, String direction) {
        
		int[] answer = new int[numbers.length];

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