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

📎 문제 출처

---

https://school.programmers.co.kr/learn/courses/30/lessons/181894

📌 문제 설명

---

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

❓ 풀이 방법

---

처음 등장하는 2의 index와 마지막으로 등장하는 2의 index를 찾아서 두 index 사이의 배열을 copyOfRange로 복사해서 리턴한다.

📌 Code

---

import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
        int firstIndex = -1;
        int lastIndex = -1;
        
        for(int i = 0; i < arr.length; i++) {
            if(arr[i] == 2) {
                firstIndex = i;
                break;
            }
        }
        
        for(int i = arr.length-1; i >= 0; i--) {
            if(arr[i] == 2) {
                lastIndex = i;
                break;
            }
        }
        
        if(firstIndex != -1 && lastIndex != -1) {
            return Arrays.copyOfRange(arr, firstIndex, lastIndex + 1);
        }
        
        return new int[] {-1};
    }
}