[프로그래머스-코딩 기초 트레이닝] 2의 영역

Hi, There!
안녕하세요, 바오밥입니다.

---

목차

• 문제
• 풀이

 

---

문제

문제 내용

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

프로그래머스

---

풀이

나의 풀이

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int[] solution(int[] arr) {
        ArrayList<Integer> idxList = new ArrayList<>();
        ArrayList<Integer> intList = new ArrayList<>();
        
        for(int i=0; i<arr.length; i++)
            if(arr[i]==2) 
                idxList.add(i);

        if(idxList.size() == 1) intList.add(2);
        if(idxList.size() == 0) intList.add(-1);
        if(idxList.size() > 1) {
            for(int i=Collections.min(idxList); i<=Collections.max(idxList); i++) 
                intList.add(arr[i]);
        }

        return intList.stream().mapToInt(x->x).toArray();
    }
}

 

다른 사람의 풀이

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr) {
        int[] idx = IntStream.range(0, arr.length).filter(i -> arr[i] == 2).toArray();
        if (idx.length == 0) return new int[]{-1};
        return IntStream.rangeClosed(idx[0], idx[idx.length - 1]).map(i -> arr[i]).toArray();
    }
}