[프로그래머스] 59. 2의 영역

입출력 예

입출력 예 설명

• 2가 있는 인덱스는 1번, 5번 인덱스뿐이므로 1번부터 5번 인덱스까지의 부분 배열인 [2, 1, 4, 5, 2]를 return 합니다.

• 2가 한 개뿐이므로 [2]를 return 합니다.

• 2가 배열에 없으므로 [-1]을 return 합니다.

• 2가 있는 인덱스는 1번, 3번, 6번 인덱스이므로 1번부터 6번 인덱스까지의 부분 배열인 [2, 1, 2, 1, 10, 2]를 return 합니다.

코드

class Solution {
    public int[] solution(int[] arr) {
        int firstIdx = 0;
        int lastIdx = 0;
        int[] result = {-1}; // 기본값: 2가 없는 경우
        for(int i = 0; i < arr.length; i++) {
            if(arr[i] == 2) {
                firstIdx = i;
                break;
            }
        }
        for(int i = arr.length -1; i > -1; i--) {
            if(arr[i] == 2) {
                lastIdx = i;
                break;
            }
        }
        if(lastIdx == 0 && firstIdx == 0) return result;
        int[] result2 = new int[lastIdx-firstIdx+1];
        for(int i = firstIdx, j = 0; i <= lastIdx; i++, j++) {
            result2[j] = arr[i];
        }
        return result2.length > 0 ? result2 : result;
    }
}