[프로그래머스 / 자바] - 배열 만들기 3

배열 만들기 3

문제

• 정수 배열 arr과 2차원 배열 intervals가 매개변수로 주어집니다.
• intervals는 항상 [[a1, b1], [a2, b2]]의 형태로 주어집니다.
• 각 구간은 닫힌 구간이며,
• 닫힌 구간 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.
• 이때, 배열 arr의 첫 번째 구간에 해당하는 배열과
• 두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 출력하는 문제입니다.
• ex)
  intervals 1번 원소 1 ~ 3 구간의 arr 원소들 2, 3,
  intervals 2번 원소 0 ~ 4 구간의 arr 원소들 1, 2, 3, 4, 5
  위 두 개를 붙여서 배열로 출력 == [2, 3, 4, 1, 2, 3, 4, 5]

• arr	intervals	result
  [1, 2, 3, 4, 5]	[[1, 3], [0, 4]]	[2, 3, 4, 1, 2, 3, 4, 5]

---

테스트 (인텔리제이)

• 우선 intervals의 각 원소들을 추출하기 위해 2중 for문을 사용했습니다.
• 첫 번째 for문은 intervals의 크기 만큼 순회하도록,
• 두 번째 for문은 intervals의 각 원소 안에 있는 원소를 추출해야 하기 때문에 첫 번째 for문의 i값을 첫 번째 대괄호에 두 번째 대괄호에는 0 ~ 1로 지정해줍니다.
• 이제 List로 선언해둔 answer에 하나씩 추가해줍니다.

import java.util.ArrayList;
import java.util.List;

public class 배열_만들기_3 {
  public static void main(String[] args) {
    /*
    정수 배열 arr과 2차원 배열 intervals가 매개변수로 주어집니다.
    intervals는 항상 [[a1, b1], [a2, b2]]의 형태로 주어집니다.
    각 구간은 닫힌 구간이며,
    닫힌 구간 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.
    이때, 배열 arr의 첫 번째 구간에 해당하는 배열과
    두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 출력하는 문제입니다.
    ex) intervals 1번 원소 1 ~ 3 구간의 arr 원소들 2, 3, 4
        intervals 2번 원소 0 ~ 4 구간의 arr 원소들 1, 2, 3, 4, 5
        위 두 개를 붙여서 배열로 출력 == [2, 3, 4, 1, 2, 3, 4, 5]
    arr	              intervals	          result
    [1, 2, 3, 4, 5]	  [[1, 3], [0, 4]]	  [2, 3, 4, 1, 2, 3, 4, 5]
     */

    int[] arr = {1, 2, 3, 4, 5};
    int[][] intervals = {{1, 3}, {0, 4}};
    List<Integer> answer = new ArrayList<>();
    for (int i = 0; i < intervals.length; i++) {
      for (int j = intervals[i][0]; j <= intervals[i][1]; j++) {
        answer.add(arr[j]);
      }
    }

    System.out.println(answer);
  }
}

---

프로그래머스 코드

import java.util.*;

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