[코딩 기초 트레이닝] 2의 영역

문제 :

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

프로그래머스

 

해결 :

import java.util.*;
class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        
        for(int i =0; i<arr.length; i++){
            if(arr[i] == 2){
                list.add(i);
            }            
        }
        
        if(list.size() == 0){
            int[] answer = {-1};
            return answer;
        }else{
            return Arrays.copyOfRange(arr, list.get(0), list.get(list.size()-1)+1);
        }
        
    }
}