Basic70 수열과 구간 쿼리 1

문제: 수열과 구간 쿼리 1

문제: 수열과 구간 쿼리 1

문제 설명

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e] 꼴입니다.

각 query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 arr[i]에 1을 더합니다.

위 규칙에 따라 queries를 처리한 이후의 arr를 return 하는 solution 함수를 완성해 주세요.

---

제한사항

• 1 ≤ arr의 길이 ≤ 1,000
  • 0 ≤ arr의 원소 ≤ 1,000,000
• 1 ≤ queries의 길이 ≤ 1,000
  • 0 ≤ s ≤ e < arr의 길이

---

입출력 예

arr	queries	result
[0, 1, 2, 3, 4]	[[0, 1],[1, 2],[2, 3]]	[1, 3, 4, 4, 4]

---

입출력 예 설명

입출력 예 #1

• 각 쿼리에 따라 arr가 다음과 같이 변합니다.

i	queries[i]	arr
-	-	[0, 1, 2, 3, 4]
0	[0,1]	[1, 2, 2, 3, 4]
1	[1,2]	[1, 3, 3, 3, 4]
2	[2,3]	[1, 3, 4, 4, 4]

• 따라서 [1, 3, 4, 4, 4]를 return 합니다.

---

Solution

먼저 queries.reduce(into: arr)을 사용해서 arr을 reduce의 초기값으로 넣어주었습니다.

queries.reduce(into: arr){ result, q in }

그런 다음 arr.indices.filter를 사용해서 index의 값이 queries의 범위에 있는 index들을 뽑아낸 다음 그에 해당하는 요소에 += 1를 해주는 방법으로 문제를 풀었습니다.

import Foundation

func solution(_ arr:[Int], _ queries:[[Int]]) -> [Int] {
    return queries.reduce(into: arr){ result, q in
        arr.indices.filter{$0 >= q[0] && $0 <= q[1]}.map{result[$0] += 1}
    }
}