● 문제
1
2
3
정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다.
배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을
return하도록 solution 함수를 완성해주세요.
● 제한사항
1
2
1) 3 ≤ numbers의 길이 ≤ 20
2) direction은 "left" 와 "right" 둘 중 하나입니다.
● 입출력 예
numbers | direction | result |
---|---|---|
[1, 2, 3] | “right” | [3, 1, 2] |
[4, 455, 6, 4, -1, 45, 6] | “left” | [455, 6, 4, -1, 45, 6, 4] |
● 코드
1. 일반 배열
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution {
public int[] solution(int[] numbers, String direction) {
int[] answer = new int[numbers.length];
int len = answer.length;
if (direction.equals("right")) {
answer[0] = numbers[len - 1];
for(int i=1; i<len; i++) {
answer[i] = numbers[i-1];
}
} else {
answer[len - 1] = numbers[0];
for(int i=0; i<len - 1; i++) {
answer[i] = numbers[i+1];
}
}
return answer;
}
}
2. ArrayList
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
class Solution {
public int[] solution(int[] numbers, String direction) {
List<Integer> list = new ArrayList<Integer>();
int[] answer = new int[numbers.length];
int len = answer.length;
if (direction.equals("right")) {
list.add(numbers[len - 1]);
for(int i=1; i<len; i++) {
list.add(numbers[i-1]);
}
} else {
for(int i=0; i<len - 1; i++) {
list.add(numbers[i+1]);
}
list.add(numbers[0]);
}
for(int i=0; i<answer.length; i++) {
answer[i] = list.get(i);
}
return answer;
}
}