[TIL] 2023.06.13 Programmers_수열과 구간 쿼리 1

수열과 구간 쿼리 1

문제 설명

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

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

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

 

 

입출력 예

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]

 

 

나의 풀이

function solution(arr, queries) {
    let answer = arr;
    for (const query of queries) {
        const [s, e] = query;
        for (const i in arr) {
            if (s <= i && e >= i) {
                arr[i] += 1;
            };
        };
    };
    return answer;
}

1. queries에 for문을 사용해서 각 요소를 s, e로 구조분해할당한다.
2. 이중 for문을 사용해서 arr의 index가 s와 s 사이에 속한다면 각 요소에 +1을 한다.
3. 최종적으로 answer를 return 한다.

 

 

다른 사람의 풀이 01

function solution(arr, queries) {
    queries.forEach(([s, e]) => {
        while (s <= e) arr[s++]++;
    });
    return arr;
}

while문을 이용해서 s가 e보다 작거나 같다면 s에 1씩 더해가면서 arr의 index에 접근해 arr의 요소까지 1 더해준다.

 

 

다른 사람의 풀이 02

function solution(arr, queries) {
    for (let [s,e] of queries) for (let i=s; i<= e; i++) arr[i]++
    return arr
}

두 번째 for문의 조건문 자체를 s와 e를 이용해서 설정하면 for문 안에 따로 if문을 추가할 필요가 없다.