프로그래머스 - java 2의 영역

💡  좋은 코드 !

배열 안의 2가 모두 포함된 "가장 작은 연속된 " 부분 배열을 리턴하는 문제이다.

먼저 arr 만큼 돌면서 처음 2가 나오는 숫자를 start 에 넣어준다.

이후 인덱스를 역순으로 돌리면서 마지막 2 가 나오는 부분을 찾아준다

(이 부분이 막혔는데 되게 간단하게 맨 뒤 2를 찾으면 되는 거였다)

 

그리고 start 값과 end 값이 지정되지 않으면 => 2가 없다는 뜻이므로 -1을 삽입해주고

else  > start 에서 end  인덱스까지 arr 값을 삽입해주면 된다 !!

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;
    }
}

 

 

실패한 코드 ㅜㅜ

import java.util.ArrayList;

class Solution {
    public ArrayList <Integer> solution(int[] arr) {
          ArrayList <Integer> answer = new ArrayList <Integer>();
         int index  = 0;
        // 가장 작은 2를 찾는다
        for (int i = 0 ; i < arr.length; i ++){
            if (arr[i] == 2){
                index = i;
                answer.add(arr[i]);
                break;
            }
        }
        for (int j = index+1; j <arr.length; j++){ // 첫 2 부터 시작해서 출력하고
            answer.add(arr[j]);
            if (arr[j] == 2){  // 그다음 2를 발견하면 종료한다.
                break;
            }
            
        }
        
        return answer;
    }
}