[프로그래머스(Java)] 배열 만들기 3 / IntStream.range

 

Level. 0

 

문제

정수 배열 arr와 2개의 구간이 담긴 배열 intervals가 주어집니다.

intervals는 항상 [[a1, b1], [a2, b2]]의 꼴로 주어지며 각 구간은 닫힌 구간입니다.
닫힌 구간은 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.

이때 배열 arr의 첫 번째 구간에 해당하는 배열과 두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 return 하는 solution 함수를 완성해 주세요.

* 제한사항
- 1 ≤ arr의 길이 ≤ 100,000
   1 ≤ arr의 원소 < 100
- 1 ≤ a1 ≤ b1 < arr의 길이
- 1 ≤ a2 ≤ b2 < arr의 길이

 

 

풀이1. stream

import java.util.Arrays;
import java.util.ArrayList;
import java.util.stream.*;

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        ArrayList<Integer> list = new ArrayList<>();
        for(int i=0; i<intervals.length; i++){
            for(int j = intervals[i][0]; j<=intervals[i][1]; j++){
                list.add(arr[j]);
            }
        }

        return IntStream.range(0, list.size())
                    .map(i -> list.get(i))
                    .toArray();

        /*
        int[] answer = new int[list.size()];
        for(int i=0;i<list.size(); i++){
            answer[i] = list.get(i);
        }
        */
        return answer;
    }
}

 

list를 array로 변경하는 과정에서 stream을 사용하였는데 (단순히 stream을 응용해보고 싶었다!)

속도면에서는 stream API를 사용하는 것 보다 forloop 가 더 빠르다고 한다.

 

그래서 stream말고 for문을 사용하여 풀이도 해보았다. 

 

 

풀이 2. forloop

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        ArrayList<Integer> list = new ArrayList<>();
        for(int i=0; i<intervals.length; i++){
            for(int j = intervals[i][0]; j<=intervals[i][1]; j++){
                list.add(arr[j]);
            }
        }

        int[] answer = new int[list.size()];
        for(int i=0;i<list.size(); i++){
            answer[i] = list.get(i);
        }
        
        return answer;
    }
}

 

- stream

 

- forloop

 

확실히 forloop가 빠르긴 하다!

그래도 stream에 익숙해지고 싶기 때문에 이후에도 간단한 문제풀이에 사용해볼 생각이다. 

 

 

---

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

프로그래머스