[programmers] 2의 영역 - Java

문제 설명

제한사항

• 1 ≤ arr의 길이 ≤ 100,000
• 1 ≤ arr의 원소 ≤ 10

입출력 예

입출력 예 설명

• 2가 있는 인덱스는 1번, 5번 인덱스뿐이므로 1번부터 5번 인덱스까지의 부분 배열인 [2, 1, 4, 5, 2]를 return 합니다.

• 2가 한 개뿐이므로 [2]를 return 합니다.

• 2가 배열에 없으므로 [-1]을 return 합니다.

• 2가 있는 인덱스는 1번, 3번, 6번 인덱스이므로 1번부터 6번 인덱스까지의 부분 배열인 [2, 1, 2, 1, 10, 2]를 return 합니다.

solution.java

import java.util.Arrays;

class Solution {
    public int[] solution(int[] arr) {
        // 초기값으로 배열의 길이보다 큰 값을 가진 변수를 설정
        int firstIndex = 100000, lastIndex = 0;
        // 배열을 순회하면서 숫자 2가 나타나는 첫 인덱스와 마지막 인덱스를 찾음
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] == 2) {
                firstIndex = Math.min(firstIndex, i);
                lastIndex = Math.max(lastIndex, i);
            }
        }
        // 만약 숫자 2가 배열 안에 존재한다면 해당 부분 배열을 반환 
        if (firstIndex <= lastIndex) {
            return Arrays.copyOfRange(arr, firstIndex, lastIndex + 1);
        } else {    //  그렇지 않다면 {-1}을 반환
            return new int[]{-1};
        }
    }
}

핵심 키워드

• Math.min() 메소드와 Math.max() 메소드를 통해 배열을 순회하며 숫자 2가 나타나는 인덱스를 저장할 수 있다.

• Arrays.copyOfRange() 메소드는 대상 배열의 지정된 범위를 새 배열로 복사한다. 첫 번째 파라미터는 소스 배열, 두 번째 파라미터는 복사할 범위의 시작 인덱스, 세 번째 파라미터는 복사할 범위의 마지막 인덱스이다.

결론!