[프로그래머스] 58. 배열 만들기 3

입출력 예

입출력 예 설명

• 첫 번째 구간에 해당하는 배열은 [2, 3, 4] 입니다.

• 두 번째 구간에 해당하는 배열은 [1, 2, 3, 4, 5] 입니다.

• 따라서 이 두 배열을 앞뒤로 붙인 배열인 [2, 3, 4, 1, 2, 3, 4, 5]를 return 합니다.

코드

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        int ff = intervals[0][0];
        int fs = intervals[0][1];
        int sf = intervals[1][0];
        int ss = intervals[1][1];
        
        int[] first = new int[fs - ff +1];
        int[] second = new int[ss - sf +1];
        
        for(int i = 0, j = ff; j <= fs; i++, j++) {
            first[i] = arr[j];
        }
        for(int i = 0, j = sf; j <= ss; i++, j++) {
            second[i] = arr[j];
        }
        
        int[] result = new int[first.length + second.length];
        
        for(int i = 0; i < first.length; i++) {
            result[i] = first[i];
        }
        for(int i = 0; i < second.length; i++) {
            result[first.length + i] = second[i];
        }
        
        return result;
    }
}

import java.util.*;
import java.util.stream.*;

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        return IntStream.concat(
                Arrays.stream(arr, intervals[0][0], intervals[0][1] + 1),
                Arrays.stream(arr, intervals[1][0], intervals[1][1] + 1)
        ).toArray();
    }
}