[JAVA/프로그래머스] 가위 바위 보

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

프로그래머스

 

문제 설명
정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

제한사항
1 ≤ arr의 길이 ≤ 100,000
1 ≤ arr의 원소 ≤ 10

입출력 예

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

입출력 예 #2
2가 한 개뿐이므로 [2]를 return 합니다.

입출력 예 #3
2가 배열에 없으므로 [-1]을 return 합니다.

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

---

👉 풀이과정

제일 처음에 있는 2와 마지막 2 사이에 있는 배열 값을 모두 출력해야 한다.

2가 몇 개 있는지 모르니까 배열 대신 리스트를 선언해서 사용한다.

반복문으로 2가 있을 경우, 리스트에 인덱스 값을 넣는다.

리스트 크기가 0인 경우에는 -1을 넣고,

아닐때는 copyOfRanger()를 사용해서 리스트 0번째 값과 마지막 값을 복사한다!!

 

📌 제출코드

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