[프로그래머스 코딩테스트] java Lv.0 2의 영역

 

 

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