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

문제

문자열 리스트 str_list에는 "u", "d", "l", "r" 네 개의 문자열이 여러 개 저장되어 있습니다. str_list에서 "l"과 "r" 중 먼저 나오는 문자열이 "l"이라면 해당 문자열을 기준으로 왼쪽에 있는 문자열들을 순서대로 담은 리스트를, 먼저 나오는 문자열이 "r"이라면 해당 문자열을 기준으로 오른쪽에 있는 문자열들을 순서대로 담은 리스트를 return하도록 solution 함수를 완성해주세요. "l"이나 "r"이 없다면 빈 리스트를 return합니다.

- 제한사항

• 1 ≤ str_list의 길이 ≤ 20
• str_list는 "u", "d", "l", "r" 네 개의 문자열로 이루어져 있습니다.

풀이

직접 푼 문제 중에 최고점 획득 !!

for문을 사용하면 인덱스가 0에서 순차적으로 증가하기 때문에 먼저 나오는 문자열이라는 조건을 따로 신경 쓸 필요가 없었다. l 또는 r이기만 하면 조건에 맞게 Arrays.copyOfRange()를 통해 배열을 복사한 뒤에 break;로 루프를 빠져 나오면 되니까!

l이나 r이 없거나, 왼쪽 오른쪽에 원소가 없을 경우 Arrays.copyOfRange()가 복사해줄 배열의 인덱스 범위가 같아지는 셈이다. 이에 따라 빈 배열이 전달되기 때문에 그 또한 따로 계산해줄 필요가 없었다.

import java.util.Arrays;

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

결과