Algorithm/프로그래머스

[알고리즘] 배열 회전시키기 Java

rw- 2023. 7. 5. 22:22
728x90

 

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int[] numbers, String direction) {
        List<Integer> answer = new ArrayList<>();
        if (direction.equals("right")) {
          answer.add(numbers[numbers.length-1]);
          for (int i = 0; i < numbers.length - 1; i++) {
            answer.add(numbers[i]);
          }
        } else {
          for (int i = 1; i < numbers.length; i++) {
            answer.add(numbers[i]);
          }
          answer.add(numbers[0]);
        }
        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}
728x90
반응형