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

문제출처

프로그래머스

---

문제

 

입출력 예

 

소스코드1 - List 사용

import java.util.*;
class Solution {
    public int[] solution(int[] arr) {
        List<Integer> list = new ArrayList<>();
        
        // 배열을 순회하면서 2가 포함된 모든 idx를 list에 저장
        for(int i=0; i<arr.length; i++){
            if(arr[i]==2){
                list.add(i); 
            }
        }
        
        if(!list.isEmpty()){    // 리스트에 값이 존재
            int min = list.get(0);  // 가장 첫번째 idx - 가장 작은 idx
            int max = list.get(list.size()-1);  // 가장 마지막 idx - 가장 큰 idx         
            return Arrays.copyOfRange(arr, min, max + 1);   // 배열 복사 메서드                              
        }else{  // 리스트에 값이 존재하지 않을 때
            return new int[]{-1};
        }
        
    }
}

 

풀이

arr 배열을 순회하면서 2가 포함된 모든 idx를 list에 저장 후 첫번째 idx, 마지막 inx를 max, min 변수에 담아준다.
Arrays.copyOfRange() - 배열 복사 메서드를 통해 min부터 max까지의 배열을 복사후 리턴해준다. 리스트에 값이 존재하지 않는다면 -1을 리턴해준다.

---

소스코드2 - List 사용 X

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr) {
        int min = 100000, max = 0;
        // 배열을 순회하면서 값이 2인 가장 작은 idx, 가장 큰 idx를 변수에 담아준다.
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] == 2) {
                min = Math.min(min, i);
                max = Math.max(max, i);
            }
        }
        // min<=max -> 배열 안에 2가 존재 할 때
        if (min <= max) {
            return Arrays.copyOfRange(arr, min, max + 1);
        } else {
            return new int[]{-1};
        }
    }
}

 

풀이

처음 풀이때는 List를 선언 후 저장된 값을 불러와 첫번째 idx, 마지막 idx를 추출했는데, List를 사용하지않아도
배열을 순회하면서 첫번째 idx(min) 값과 마지막 idx(max)값을 추출할 수 있어 새로운 풀이를 시도해봤습니다.