[JAVA] 프로그래머스 - 2의 영역

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

프로그래머스

 

 

 

 

 

 

정답 

 

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr) {
        
        ArrayList<Integer> temp = new ArrayList<>();
        
        
        for(int i=0; i<arr.length; i++)
        {
            if(arr[i]==2)
            {
                temp.add(i); //값이 2인 인덱스 리스트에 추가 
            }
        }
        
        if(temp.size()==0) {
            int[] answer1 = {-1};
            return answer1;
        }
        else
        {
            int start = temp.get(0);
            int end = temp.get(temp.size()-1);
        
            int answer[] = new int[end-start+1];
            int idx=0;
            for(int i=start; i<=end; i++)
            {
                answer[idx++] = arr[i];
            }
            
            return answer;
        }
        
    }
}

 

 

 

다른 사람의 풀이 

 

 

import java.util.*;
class Solution {
    public ArrayList<Integer> solution(int[] arr) {
        ArrayList<Integer> answer = new ArrayList<Integer>();
        int start = -1;
        int end = -1;
        for(int i = 0; i<arr.length; i++){
            if(arr[i]==2){
                start = i;
                break;
            }
        }
        for(int i = arr.length-1; i>=0; i--){
            if(arr[i]==2){
                end = i;
                break;
            }
        }
        if(start==-1 && end==-1){
            answer.add(-1);
        } else{
            for(int i = start; i<=end; i++){
                answer.add(arr[i]);
            }
        }

        return answer;
    }
}