[프로그래머스] Lv.0 왼쪽 오른쪽.java

왼쪽 오른쪽.java

import java.util.Arrays;

class Solution {
    public String[] solution(String[] str_list) {
        String[] answer = {};
        for (int i = 0; i < str_list.length;i++) {
            if (str_list[i].equals("l")) {
                return Arrays.copyOfRange(str_list, 0, i);
            } else if (str_list[i].equals("r")) {
                return Arrays.copyOfRange(str_list, i + 1, str_list.length);
            }
        }
        return answer;
    }
}

현재 요소가 "l"인 경우, "l"이 처음으로 나왔으므로, Arrays.copyOfRange 메서드를 사용하여 str_list 배열의 0부터 i-1까지의 부분 배열을 반환합니다.
"r"인 경우, "r"이 처음으로 나왔으므로, Arrays.copyOfRange 메서드를 사용하여 str_list 배열의 i+1부터 끝까지의 부분 배열을 반환합니다.
만약 "l"이나 "r"이 발견되지 않았다면, 초기에 생성한 빈 배열 answer를 반환합니다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges