Basic58 배열 만들기 3

문제: 배열 만들기 3

문제 설명

정수 배열 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의 길이

---

입출력 예

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

---

입출력 예 설명

입출력 예 #1

• 첫 번째 구간에 해당하는 배열은 [2, 3, 4] 입니다.
• 두 번째 구간에 해당하는 배열은 [1, 2, 3, 4, 5] 입니다.
• 따라서 이 두 배열을 앞뒤로 붙인 배열인 [2, 3, 4, 1, 2, 3, 4, 5]를 return 합니다.

---

Solution

intervals에 map을 접근한 요소의 값으로 arr을 나누어서 각각 intervals의 값이 요구하는 배열을 생성합니다. 입출력 예를 입력으로 print해보면 아래와 같습니다. [ArraySlice([2, 3, 4]), ArraySlice([1, 2, 3, 4, 5])] 이제 이 배열들을 reduce를 사용해서 하나의 배열로 만들었습니다. reduce의 $0으로 들어가게 되는 ()안의 값이 [] 빈 배열이 되면 배열끼리 더하는 코드를 작성할 수 있게 됩니다.

import Foundation

func solution(_ arr:[Int], _ intervals:[[Int]]) -> [Int] {
    return intervals.map{arr[$0[0]...$0[1]]}.reduce([]){$0+$1}
}