[0단계] 2의 영역

점수 : 4점 

 

import java.util.*;
class Solution {
    public int[] solution(int[] arr) {
     
        List<Integer> two = new ArrayList<>(); //arr[i] 가 2인 인덱스 담기
        
        for(int i =0; i<arr.length;i++){
            if(arr[i]==2){
                two.add(i);
            }
        }
     
              
            if(two.isEmpty()){    //2가 없다면 -1
            return new int[]{-1};
        }
        
        //2가 있다면 
        int s= two.get(0); 
        int e =two.get(two.size()-1); 
        int[] answer = new int [e-s+1]; 
            
         
        for(int i =s, index = 0; i<e+1; i++,index++){
                answer[index] = arr[i];
                 
            }
        
        return answer;
    }
}