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

문제출처

프로그래머스

---

문제

 

입출력 예

 

소스코드

import java.util.*;
class Solution {
    public List<Integer> solution(int[] arr, int[][] intervals) {
        List<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 list;
    }
}

 

풀이

이중 for문을 사용해 선언한 list에 저장해주고 return해준다.

---

소스코드2

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        int size = intervals[0][1]-intervals[0][0]+intervals[1][1]-intervals[1][0]+2;
        int[] answer = new int[size];
        int cnt = 0;
        
        for(int i=0; i<intervals.length; i++){
            for(int j=intervals[i][0]; j<=intervals[i][1]; j++){
                answer[cnt] = arr[j];
                cnt++;
            }
        }

        return answer;
    }
}

 

풀이

List를 선언하지않고 풀수없을까 생각해보다가 int 배열의 길이를 계산해 선언후 배열에 값을 담아 int 배열 그대로 return했습니다.